hendrydong commited on
Commit
d707bd7
·
1 Parent(s): 0981bb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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 = "pinkmanlove/llama-7b-hf"
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