Update run_Transformers_imgunderstand.py

LLaDA2.0/run_Transformers_imgunderstand.py

@@ -6,7 +6,7 @@ from decoder.smart_img_process import smart_resize_images
 model_path = "inclusionAI/LLaDA2.0-Uni"
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
-    model_path, device_map="cuda", torch_dtype="bfloat16", trust_remote_code=True
+    model_path, device_map="auto", torch_dtype="bfloat16", trust_remote_code=True
 ).eval()
 model.tokenizer = tokenizer