From e1000551f05c8afba5ca86b9ce05bb6989289f5e Mon Sep 17 00:00:00 2001
From: Asia <2736300+humpydonkey@users.noreply.github.com>
Date: Tue, 20 Feb 2024 00:07:40 +0800
Subject: [PATCH] [FIX]Fix a NameError for llama-index-embeddings-clip (#10974)
---
.../llama_index/embeddings/clip/base.py | 2 ++
1 file changed, 2 insertions(+)
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 6bc6da46c..e00840128 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))
--
GitLab