refactor: add imports

modeling_clip.py

@@ -5,8 +5,9 @@
 # and adjusted for Jina CLIP
 
 from functools import partial
-from typing import Optional, Tuple, Union
+from typing import Optional, Tuple, Union, List
 
+import numpy as np
 import torch
 import torch.nn.functional as f
 import torch.utils.checkpoint
@@ -258,7 +259,7 @@ class JinaCLIPModel(JinaCLIPPreTrainedModel):
         device: Optional[torch.device] = None,
         normalize_embeddings: bool = False,
         **tokenizer_kwargs,
-    ) -> Union[Tuple[Optional[torch.FloatTensor], ...], CLIPTextModelOutput]:
+    ) -> Union[List[torch.Tensor], np.ndarray, torch.Tensor]::
         
         self.eval()