bwang0911 commited on
Commit
2291b67
·
verified ·
1 Parent(s): e30e2df

Update modeling_clip.py

Browse files
Files changed (1) hide show
  1. 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.process([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)
 
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)