Merge pull request #1188 from ncoop57/clip-image-check

Change instance check to image instead of imagefile
UKPLab · Oct 1, 2021 · ffd72ab · ffd72ab
2 parents 75c6b2f + 603df3c
commit ffd72ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sentence_transformers/models/CLIPModel.py b/sentence_transformers/models/CLIPModel.py
@@ -1,7 +1,7 @@
 from torch import nn
 import transformers
 import torch
-from PIL import ImageFile
+from PIL import Image
 
 
 class CLIPModel(nn.Module):
@@ -56,7 +56,7 @@ def tokenize(self, texts):
         image_text_info = []
 
         for idx, data in enumerate(texts):
-            if isinstance(data, ImageFile.ImageFile):  # An Image
+            if isinstance(data, Image.Image):  # An Image
                 images.append(data)
                 image_text_info.append(0)
             else:  # A text