Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
•
b48d5f8
1
Parent(s):
53518f7
自动从HG下载llama模型
Browse files- 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 |
-
|
|
|
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
|