Update modeling_clip.py
Browse files- modeling_clip.py +1 -1
modeling_clip.py
CHANGED
@@ -404,7 +404,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
|
|
404 |
If convert_to_tensor, a stacked tensor is returned.
|
405 |
If convert_to_numpy, a numpy matrix is returned.
|
406 |
"""
|
407 |
-
from PIL
|
408 |
|
409 |
is_training = self.training
|
410 |
self.eval()
|
|
|
404 |
If convert_to_tensor, a stacked tensor is returned.
|
405 |
If convert_to_numpy, a numpy matrix is returned.
|
406 |
"""
|
407 |
+
from PIL import Image
|
408 |
|
409 |
is_training = self.training
|
410 |
self.eval()
|