Spaces:
Runtime error
Runtime error
Commit
·
d707bd7
1
Parent(s):
0981bb9
Update app.py
Browse files
app.py
CHANGED
|
@@ -107,8 +107,7 @@ def main():
|
|
| 107 |
model_args, pipeline_args, chatbot_args = (
|
| 108 |
parser.parse_args_into_dataclasses()
|
| 109 |
)
|
| 110 |
-
model_args.model_name_or_path = "
|
| 111 |
-
model_args.lora_model_path = "./robin-7b"
|
| 112 |
pipeline_args.deepspeed = "configs/ds_config_chatbot.json"
|
| 113 |
model_args.torch_dtype = "float16"
|
| 114 |
|
|
|
|
| 107 |
model_args, pipeline_args, chatbot_args = (
|
| 108 |
parser.parse_args_into_dataclasses()
|
| 109 |
)
|
| 110 |
+
model_args.model_name_or_path = "LMFlow/Robin-7b-v2"
|
|
|
|
| 111 |
pipeline_args.deepspeed = "configs/ds_config_chatbot.json"
|
| 112 |
model_args.torch_dtype = "float16"
|
| 113 |
|