fix: llama-3

src/model/model_llava.py

@@ -114,7 +114,7 @@ def inference_by_prompt_and_images_fire(prompt, images):
     with torch.inference_mode():
         cont = model_llava_fire.generate(
             input_ids,
-            images=image_tensor,
+            images=[image_tensor.squeeze(dim=0)],
             image_sizes=image_sizes,
             do_sample=False,
             temperature=0,