dj86 commited on
Commit
528ff68
1 Parent(s): df621a3

Update vlog4chat.py

Browse files
Files changed (1) hide show
  1. vlog4chat.py +2 -2
vlog4chat.py CHANGED
@@ -158,8 +158,8 @@ class Vlogger4chat :
158
  #self.vector_storage = FAISS.from_documents(chunks, self.my_embedding)
159
  pkl_path = os.path.join(self.data_dir, f"{video_id}.pkl")
160
  with open(pkl_path, 'rb') as f:
161
- vector_storage =pickle.load(f)
162
- self.vector_storage = map_to_cpu(vector_storage)
163
  self.chain = ConversationalRetrievalChain.from_llm(self.llm, self.vector_storage.as_retriever(), return_source_documents=True)
164
  return True
165
  return False
 
158
  #self.vector_storage = FAISS.from_documents(chunks, self.my_embedding)
159
  pkl_path = os.path.join(self.data_dir, f"{video_id}.pkl")
160
  with open(pkl_path, 'rb') as f:
161
+ self.vector_storage = pickle.load(f)
162
+
163
  self.chain = ConversationalRetrievalChain.from_llm(self.llm, self.vector_storage.as_retriever(), return_source_documents=True)
164
  return True
165
  return False