Merge pull request #254 from hill2hill/main

fix: eval when generate_with_interleaved
This commit is contained in:
Cui Junbo
2024-06-13 03:21:05 +08:00
committed by GitHub
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=None,
msgs=msgs,
context=None,
tokenizer=self.tokenizer,