jinaai
/

jina-clip-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

hanxiao commited on May 30

Commit

6cf8056

•

1 Parent(s): 806a44b

Update modeling_clip.py

Files changed (1) hide show

modeling_clip.py +1 -1

modeling_clip.py CHANGED Viewed

@@ -391,7 +391,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
     @torch.inference_mode()
     def encode_image(
         self,
-        images: Union[str, List[Union[str, Image.Image]]],
         batch_size: int = 32,
         show_progress_bar: Optional[bool] = None,
         convert_to_numpy: bool = True,

     @torch.inference_mode()
     def encode_image(
         self,
+        images: Union[str, List[Union[str, "Image.Image"]]],
         batch_size: int = 32,
         show_progress_bar: Optional[bool] = None,
         convert_to_numpy: bool = True,