Update README.md

README.md

@@ -12,7 +12,7 @@ import torch
 from transformers import AutoModelForCausalLM
 
 model = AutoModelForCausalLM.from_pretrained("q-future/one-align", trust_remote_code=True, 
-                                             torch_dtype=torch.float16, device_map={"":"cuda:0"})
+                                             torch_dtype=torch.float16, device_map="auto")
 
 from PIL import Image
 url = "https://raw.githubusercontent.com/Q-Future/Q-Align/main/fig/singapore_flyer.jpg"