mirror of
https://github.com/OpenBMB/MiniCPM-V.git
synced 2026-02-04 17:59:18 +08:00
fix: eval when generate_with_interleaved
This commit is contained in:
@@ -56,7 +56,7 @@ def parse_args():
|
||||
parser.add_argument("--device", type=str, default="cuda:0")
|
||||
parser.add_argument('--batchsize', type=int, default=1, help='Batch size for processing.')
|
||||
|
||||
parser.add_argument("--ckpt", type=str, default="")
|
||||
parser.add_argument("--ckpt", type=str, default=None)
|
||||
|
||||
args = parser.parse_args()
|
||||
return args
|
||||
@@ -84,6 +84,7 @@ class MiniCPM_V:
|
||||
msgs = [{'role': 'user', 'content': content}]
|
||||
|
||||
res = self.model.chat(
|
||||
image=image,
|
||||
msgs=msgs,
|
||||
context=None,
|
||||
tokenizer=self.tokenizer,
|
||||
|
||||
Reference in New Issue
Block a user