first commit
Browse files- 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)
|