[FIX]Fix a NameError for llama-index-embeddings-clip (#10974)

run-llama · Feb 19, 2024 · e100055 · e100055
1 parent 069f2c1
commit e100055
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...x-integrations/embeddings/llama-index-embeddings-clip/llama_index/embeddings/clip/base.py b/...x-integrations/embeddings/llama-index-embeddings-clip/llama_index/embeddings/clip/base.py
@@ -131,6 +131,8 @@ async def _aget_image_embedding(self, img_file_path: ImageType) -> Embedding:
         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))