From 41deae34dd68c07becbfa4b77f9890b6209db9cf Mon Sep 17 00:00:00 2001 From: HJ <92386084+JJJYmmm@users.noreply.github.com> Date: Thu, 1 Feb 2024 02:13:51 +0800 Subject: [PATCH] Update eval.py fix https://github.com/AILab-CVC/SEED-Bench/issues/21#issue-2110689739 --- eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eval.py b/eval.py index 2f1eb29..ecbbcb3 100644 --- a/eval.py +++ b/eval.py @@ -168,6 +168,6 @@ def run_inference(model, qa_anno, output_dir): print(f'evaluating.. {args.model}') # The interface for testing MLLMs - model = build_model(args.model).cuda() + model = build_model(args.model) run_inference(model, qa_anno, args.output_dir)