Update app.py

app.py

@@ -55,7 +55,7 @@ model_kwargs = {
     "torch_dtype": torch.bfloat16,
     "attn_implementation": "sdpa"
 }
-tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, device_map="cuda:0", **model_kwargs)
+tokenizer, model, image_processor, context_len = load_pretrained_model(args.model_path, args.model_base, model_name, device_map="cpu:0", **model_kwargs)
 our_chatbot = None
 
 def upvote_last_response(state):