yiyixin commited on
Commit
862bece
1 Parent(s): 58bbe62
Files changed (1) hide show
  1. embedding_model.py +1 -1
embedding_model.py CHANGED
@@ -14,7 +14,7 @@ class HuggingfaceSentenceTransformerModel(Embedding_Model):
14
  def __init__(self, model_name=EMBEDDING_MODEL) -> None:
15
  super().__init__(model_name)
16
 
17
- self.model = SentenceTransformer(model_name)
18
 
19
  def __call__(self, text) -> None:
20
  return self.model.encode(text)
 
14
  def __init__(self, model_name=EMBEDDING_MODEL) -> None:
15
  super().__init__(model_name)
16
 
17
+ self.model = SentenceTransformer(model_name, cache_folder="/app/ckpt/")
18
 
19
  def __call__(self, text) -> None:
20
  return self.model.encode(text)