Update app.py

app.py

@@ -21,7 +21,7 @@ os.environ["HUGGINGFACE_HUB_CACHE"] = os.getcwd() + "/weights"
 disable_torch_init()
 model_id = "Yanqing0327/LLaVA-project"  # 替换为你的 Hugging Face 模型仓库
 tokenizer, model, image_processor, context_len = load_pretrained_model(
-    model_id, model_name="llava-v1.5-13b", model_base=None, load_8bit=False, load_4bit=False
+    model_id, model_name="llava-v1.5-13b", model_base=None, load_8bit=False, load_4bit=False, device_map="auto"
 )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"