Update run_Transformers_t2i.py

LLaDA2.0/run_Transformers_t2i.py

@@ -5,7 +5,7 @@ from decoder import decode_vq_tokens
 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