antimoda1 commited on
Commit
96830f5
·
1 Parent(s): 3302068

add cache for embeddings

Browse files
Files changed (1) hide show
  1. retrieval.py +1 -1
retrieval.py CHANGED
@@ -83,7 +83,7 @@ class Retrieval:
83
 
84
  # Инициализируем CrossEncoder
85
  # self.cross_encoder = CrossEncoder('DiTy/cross-encoder-russian-msmarco')
86
- self.embedder = SentenceTransformer('cointegrated/rubert-tiny2')
87
  self.embeddings_of_summary = self.embedder.encode(self.paragraphs_df['summary'], convert_to_tensor=True)
88
 
89
  print("✅ RAG система готова к использованию")
 
83
 
84
  # Инициализируем CrossEncoder
85
  # self.cross_encoder = CrossEncoder('DiTy/cross-encoder-russian-msmarco')
86
+ self.embedder = SentenceTransformer('cointegrated/rubert-tiny2', cache_folder="/tmp")
87
  self.embeddings_of_summary = self.embedder.encode(self.paragraphs_df['summary'], convert_to_tensor=True)
88
 
89
  print("✅ RAG система готова к использованию")