alienet commited on
Commit
61a4198
·
1 Parent(s): a7c710f

first commit

Browse files
Files changed (1) hide show
  1. modules/embedding.py +1 -1
modules/embedding.py CHANGED
@@ -16,7 +16,7 @@ class EmbeddingModel:
16
  model_provider = model_name.split("/")[0]
17
  model_smallname = model_name.split("/")[1]
18
  model_path = os.path.join(cache_dir, f"models--{model_provider}--{model_smallname}/snapshots/")
19
- if get_child_folders(model_path):
20
  try:
21
  model_path = os.path.join(model_path,get_child_folders(model_path)[0])
22
  self.tokenizer = AutoTokenizer.from_pretrained(model_path)
 
16
  model_provider = model_name.split("/")[0]
17
  model_smallname = model_name.split("/")[1]
18
  model_path = os.path.join(cache_dir, f"models--{model_provider}--{model_smallname}/snapshots/")
19
+ if os.path.exists(model_path) and get_child_folders(model_path):
20
  try:
21
  model_path = os.path.join(model_path,get_child_folders(model_path)[0])
22
  self.tokenizer = AutoTokenizer.from_pretrained(model_path)