Bug resolved

thu-ml · Aug 8, 2024 · bc83a87 · bc83a87
1 parent 4da3606
commit bc83a87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmte/models/llava/eval/run_llava.py b/mmte/models/llava/eval/run_llava.py
@@ -177,7 +177,7 @@ def chat_model(tokenizer, model, image_processor, args):
             images,
             image_processor,
             model.config
-        ).to(model.device, dtype=torch.float16)
+        ).to(model.device, dtype=args.dtype)
     else:
         images_tensor = None
         image_sizes = None