yzm0034 commited on
Commit
6453ac2
·
verified ·
1 Parent(s): 76a5b51

model exception issue fixed

Browse files
Files changed (1) hide show
  1. 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_