Update modeling_clip.py
Browse files- modeling_clip.py +2 -0
modeling_clip.py
CHANGED
|
@@ -466,6 +466,8 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
|
|
| 466 |
)
|
| 467 |
else:
|
| 468 |
range_iter = range(0, len(images), batch_size)
|
|
|
|
|
|
|
| 469 |
|
| 470 |
for i in range_iter:
|
| 471 |
batch_images = images[i:i+batch_size]
|
|
|
|
| 466 |
)
|
| 467 |
else:
|
| 468 |
range_iter = range(0, len(images), batch_size)
|
| 469 |
+
|
| 470 |
+
from PIL import Image
|
| 471 |
|
| 472 |
for i in range_iter:
|
| 473 |
batch_images = images[i:i+batch_size]
|