hendrydong commited on
Commit
603f326
·
1 Parent(s): 4984279

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ def main():
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)
115
 
 
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)
115