diff --git a/llama-index-integrations/embeddings/llama-index-embeddings-clip/llama_index/embeddings/clip/base.py b/llama-index-integrations/embeddings/llama-index-embeddings-clip/llama_index/embeddings/clip/base.py index 6bc6da46c37fca7f2b3d446f9c332c8efe501571..e008401289c6e055c78e1fd1b48f7350b92b3ead 100644 --- a/llama-index-integrations/embeddings/llama-index-embeddings-clip/llama_index/embeddings/clip/base.py +++ b/llama-index-integrations/embeddings/llama-index-embeddings-clip/llama_index/embeddings/clip/base.py @@ -131,6 +131,8 @@ class ClipEmbedding(MultiModalEmbedding): return self._get_image_embedding(img_file_path) def _get_image_embedding(self, img_file_path: ImageType) -> Embedding: + import torch + with torch.no_grad(): image = ( self._preprocess(Image.open(img_file_path))