Tuchuanhuhuhu commited on
Commit
b48d5f8
1 Parent(s): 53518f7

自动从HG下载llama模型

Browse files
Files changed (1) hide show
  1. modules/models.py +2 -1
modules/models.py CHANGED
@@ -300,7 +300,8 @@ class LLaMA_Client(BaseLLMModel):
300
  if model_path is not None:
301
  model_source = model_path
302
  else:
303
- raise Exception(f"models目录下没有这个模型: {model_name}")
 
304
  if lora_path is not None:
305
  lora_path = f"lora/{lora_path}"
306
  self.max_generation_token = 1000
 
300
  if model_path is not None:
301
  model_source = model_path
302
  else:
303
+ model_source = f"decapoda-research/{model_name}"
304
+ # raise Exception(f"models目录下没有这个模型: {model_name}")
305
  if lora_path is not None:
306
  lora_path = f"lora/{lora_path}"
307
  self.max_generation_token = 1000