model exception issue fixed
Browse files- Models/llm_embeddings.py +2 -0
Models/llm_embeddings.py
CHANGED
|
@@ -24,6 +24,8 @@ def check_model_in_cache(model_name: str):
|
|
| 24 |
if model_name in ["olmo","OLMo"]:
|
| 25 |
return str(full_path("/data/shared/olmo/OLMo-7B_shard_size_2GB"))
|
| 26 |
|
|
|
|
|
|
|
| 27 |
def mean_pooling(model_output, attention_mask):
|
| 28 |
"""
|
| 29 |
mean_pooling _summary_
|
|
|
|
| 24 |
if model_name in ["olmo","OLMo"]:
|
| 25 |
return str(full_path("/data/shared/olmo/OLMo-7B_shard_size_2GB"))
|
| 26 |
|
| 27 |
+
raise ValueError(f"Model '{model_name}' not found in local cache.")
|
| 28 |
+
|
| 29 |
def mean_pooling(model_output, attention_mask):
|
| 30 |
"""
|
| 31 |
mean_pooling _summary_
|