Update modeling_clip.py
Browse files- modeling_clip.py +1 -1
modeling_clip.py
CHANGED
|
@@ -444,7 +444,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
|
|
| 444 |
range_iter = range(0, len(images), batch_size)
|
| 445 |
|
| 446 |
for i in range_iter:
|
| 447 |
-
processed_inputs = self.
|
| 448 |
embeddings = self.get_image_features(processed_inputs)
|
| 449 |
if normalize_embeddings:
|
| 450 |
embeddings = torch.nn.functional.normalize(embeddings, p=2, dim=1)
|
|
|
|
| 444 |
range_iter = range(0, len(images), batch_size)
|
| 445 |
|
| 446 |
for i in range_iter:
|
| 447 |
+
processed_inputs = self.preprocess([Image.open(image) for image in images])
|
| 448 |
embeddings = self.get_image_features(processed_inputs)
|
| 449 |
if normalize_embeddings:
|
| 450 |
embeddings = torch.nn.functional.normalize(embeddings, p=2, dim=1)
|