Leonydis137 commited on
Commit
20b13a3
·
verified ·
1 Parent(s): 9dc11a1

Update memory.py

Browse files
Files changed (1) hide show
  1. memory.py +1 -1
memory.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModel
4
  class MemoryVectorStore:
5
  def __init__(self, model_name="sentence-transformers/all-MiniLM-L6-v2"):
6
  self.cache_dir = "/tmp/hf_cache"
7
- os.makedirs(self.cache_dir, exist_ok=True)
8
 
9
  self.tokenizer = AutoTokenizer.from_pretrained(
10
  model_name, cache_dir=self.cache_dir
 
4
  class MemoryVectorStore:
5
  def __init__(self, model_name="sentence-transformers/all-MiniLM-L6-v2"):
6
  self.cache_dir = "/tmp/hf_cache"
7
+ os.makedirs(self.cache_dir, exist_ok=False)
8
 
9
  self.tokenizer = AutoTokenizer.from_pretrained(
10
  model_name, cache_dir=self.cache_dir