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)
|