Commit
•
ad0de50
1
Parent(s):
ed8f61a
Update model files
Browse files- processing_llava.py +1 -1
processing_llava.py
CHANGED
|
@@ -78,7 +78,7 @@ class LlavaProcessor:
|
|
| 78 |
pixel_values = self.image_processor(images, return_tensors=return_tensors)[
|
| 79 |
"pixel_values"
|
| 80 |
]
|
| 81 |
-
pixel_values = pixel_values.to(model.device)
|
| 82 |
image_outputs = model.vision_model(pixel_values)
|
| 83 |
image_features = model.multi_modal_projector(image_outputs)
|
| 84 |
else:
|
|
|
|
| 78 |
pixel_values = self.image_processor(images, return_tensors=return_tensors)[
|
| 79 |
"pixel_values"
|
| 80 |
]
|
| 81 |
+
pixel_values = pixel_values.to(model.device).to(model.dtype)
|
| 82 |
image_outputs = model.vision_model(pixel_values)
|
| 83 |
image_features = model.multi_modal_projector(image_outputs)
|
| 84 |
else:
|