fix: eval when generate_with_interleaved

This commit is contained in:
lionHC
2024-06-12 10:01:01 +08:00
committed by GitHub
parent d9cf8604f3
commit 235132f39b
2 changed files with 2 additions and 1 deletions

View File

@@ -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

View File

@@ -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,