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)
|