deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-05 02:09:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: eval when generate_with_interleaved

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
eval_mm/vqaeval/eval_utils/getargs.py
View File

@@ -56,7 +56,7 @@ def parse_args():
parser.add_argument("--device", type=str, default="cuda:0") 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('--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() args = parser.parse_args()
return args return args

1
eval_mm/vqaeval/models/MiniCPM/minicpmv.py
View File

@@ -84,6 +84,7 @@ class MiniCPM_V:
msgs = [{'role': 'user', 'content': content}] msgs = [{'role': 'user', 'content': content}]
res = self.model.chat( res = self.model.chat(
image=image,
msgs=msgs, msgs=msgs,
context=None, context=None,
tokenizer=self.tokenizer, tokenizer=self.tokenizer,