hendrydong commited on
Commit
322be28
·
1 Parent(s): edb3e83

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def main():
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.device = 'cpu'
113
  with open ("configs/ds_config_chatbot.json", "r") as f:
114
  ds_config = json.load(f)
 
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.device = 'cpu'
113
  with open ("configs/ds_config_chatbot.json", "r") as f:
114
  ds_config = json.load(f)