Update run_Transformers_imgedit.py

LLaDA2.0/run_Transformers_imgedit.py

@@ -8,7 +8,7 @@ from PIL import Image
 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