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