Spaces:

OptimalScale
/

Robin-7b

Runtime error

hendrydong commited on Jun 16, 2023

Commit

52b3bdf

1 Parent(s): d707bd7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ def main():
     model_args, pipeline_args, chatbot_args = (
         parser.parse_args_into_dataclasses()
     )
-    model_args.model_name_or_path = "LMFlow/Robin-7b-v2"
     pipeline_args.deepspeed = "configs/ds_config_chatbot.json"
     model_args.torch_dtype = "float16"

     model_args, pipeline_args, chatbot_args = (
         parser.parse_args_into_dataclasses()
     )
+    model_args.model_name_or_path = "LMFlow/Full-Robin-7b-v2"
     pipeline_args.deepspeed = "configs/ds_config_chatbot.json"
     model_args.torch_dtype = "float16"