dj86 commited on
Commit
3014fbb
1 Parent(s): cc6b261

Update vlog4chat.py

Browse files
Files changed (1) hide show
  1. vlog4chat.py +3 -1
vlog4chat.py CHANGED
@@ -25,7 +25,8 @@ from langchain.llms.base import LLM
25
  from langchain.chains import ConversationalRetrievalChain
26
  from langchain_core.output_parsers import StrOutputParser
27
  from langchain.text_splitter import RecursiveCharacterTextSplitter
28
- from langchain.llms import HuggingFacePipeline
 
29
  #from langchain.embeddings import HuggingFaceEmbeddings
30
  from langchain_community.embeddings import HuggingFaceEmbeddings
31
  from langchain_community.document_loaders import TextLoader
@@ -146,6 +147,7 @@ class Vlogger4chat :
146
  self.history = []
147
  if not os.path.exists(self.tmp_dir):
148
  os.makedirs(self.tmp_dir)
 
149
 
150
  def init_llm(self):
151
  print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')
 
25
  from langchain.chains import ConversationalRetrievalChain
26
  from langchain_core.output_parsers import StrOutputParser
27
  from langchain.text_splitter import RecursiveCharacterTextSplitter
28
+ #from langchain.llms import HuggingFacePipeline
29
+ from langchain_community.llms import HuggingFacePipeline
30
  #from langchain.embeddings import HuggingFaceEmbeddings
31
  from langchain_community.embeddings import HuggingFaceEmbeddings
32
  from langchain_community.document_loaders import TextLoader
 
147
  self.history = []
148
  if not os.path.exists(self.tmp_dir):
149
  os.makedirs(self.tmp_dir)
150
+ self.my_embedding = HuggingFaceEmbeddings(model_name='BAAI/bge-m3', model_kwargs={'device': 'cpu'} ,encode_kwargs={'normalize_embeddings': True})
151
 
152
  def init_llm(self):
153
  print('\033[1;33m' + "Initializing LLM Reasoner...".center(50, '-') + '\033[0m')