Update modeling_clip.py
Browse files- modeling_clip.py +1 -0
modeling_clip.py
CHANGED
|
@@ -490,6 +490,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
|
|
| 490 |
processed_inputs.append(image)
|
| 491 |
|
| 492 |
processed_inputs = self.preprocess(processed_inputs)
|
|
|
|
| 493 |
embeddings = self.get_image_features(processed_inputs)
|
| 494 |
|
| 495 |
if normalize_embeddings:
|
|
|
|
| 490 |
processed_inputs.append(image)
|
| 491 |
|
| 492 |
processed_inputs = self.preprocess(processed_inputs)
|
| 493 |
+
processed_inputs = processed_inputs.to(self.device)
|
| 494 |
embeddings = self.get_image_features(processed_inputs)
|
| 495 |
|
| 496 |
if normalize_embeddings:
|