Upload app.py

app.py

@@ -108,8 +108,8 @@ def main():
         parser.parse_args_into_dataclasses()
     )
     model_args.model_name_or_path = "pinkmanlove/llama-7b-hf"
-    #model_args.lora_model_path = "./robin-7b"
-    pipeline_args.device = 'cpu'
+    model_args.lora_model_path = "./robin-7b"
+    #pipeline_args.device = 'cpu'
     with open ("configs/ds_config_chatbot.json", "r") as f:
         ds_config = json.load(f)