Chris4K commited on
Commit
04d563b
1 Parent(s): cb907e8

Update vector_store_retriever.py

Browse files
Files changed (1) hide show
  1. vector_store_retriever.py +2 -2
vector_store_retriever.py CHANGED
@@ -6,7 +6,7 @@ from pydantic import BaseModel, Field
6
  from typing import Any, Optional, Dict, List, Union
7
  from huggingface_hub import InferenceClient
8
  from langchain.llms.base import LLM
9
- from langchain.Images import Images
10
  from langchain.llms.base import LLM
11
  from langchain.embeddings import HuggingFaceInstructEmbeddings, EmbeddingFunction, Embeddings
12
 
@@ -25,7 +25,7 @@ class HuggingFaceInstructEmbeddings(EmbeddingFunction):
25
  self.model = AutoModel.from_pretrained(model_name, **(model_kwargs or {}))
26
  self.tokenizer = AutoTokenizer.from_pretrained(model_name)
27
 
28
- def __call__(self, input: Union[Documents, Images]) -> Embeddings:
29
  if isinstance(input, Documents):
30
  texts = [doc.text for doc in input]
31
  embeddings = self._embed_text(texts)
 
6
  from typing import Any, Optional, Dict, List, Union
7
  from huggingface_hub import InferenceClient
8
  from langchain.llms.base import LLM
9
+ #from langchain.Images import Images
10
  from langchain.llms.base import LLM
11
  from langchain.embeddings import HuggingFaceInstructEmbeddings, EmbeddingFunction, Embeddings
12
 
 
25
  self.model = AutoModel.from_pretrained(model_name, **(model_kwargs or {}))
26
  self.tokenizer = AutoTokenizer.from_pretrained(model_name)
27
 
28
+ def __call__(self, input: Union[Documents]) -> Embeddings:
29
  if isinstance(input, Documents):
30
  texts = [doc.text for doc in input]
31
  embeddings = self._embed_text(texts)