Trying out torch.grad for speeding up

src/CLIP.py

@@ -1,4 +1,5 @@
 from transformers import AutoProcessor, CLIPModel
+import torch
 
 
 class CLIPImageEncoder:
@@ -8,6 +9,7 @@ class CLIPImageEncoder:
         self.processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
     def encode_image(self, image_pil):
-        input = self.processor(images=image_pil, return_tensors="pt")
-        image_features = self.model.get_image_features(**input)
-        return image_features.cpu().detach().numpy()[0]
+        with torch.no_grad():
+            input = self.processor(images=image_pil, return_tensors="pt")
+            image_features = self.model.get_image_features(**input)
+            return image_features.cpu().detach().numpy()[0]