Update README.md

README.md

@@ -56,9 +56,9 @@ pipe.vae=pipe.vae.to(torch.float32)
 if(enable_model_cpu_offload):
     pipe.enable_model_cpu_offload()
 else:
-    pipe.text_encoder_2=pipe.text_encoder_2.to("cuda:1")
-    pipe.transformer=pipe.transformer.to("cuda:1")
-    pipe=pipe.to("cuda:1")
+    pipe.text_encoder_2=pipe.text_encoder_2.to("cuda")
+    pipe.transformer=pipe.transformer.to("cuda")
+    pipe=pipe.to("cuda")
 
 prompt = "modern anime style, A close-up portrait of a young girl with green hair. Her hair is vibrant and shoulder-length, framing her face softly. She has large, expressive eyes that are slightly tilted upward, with a gentle and calm expression. Her facial features are delicate, with a small nose and soft lips. The background is simple, focusing attention on her face, with soft lighting that highlights her features. The overall style of the illustration is warm and inviting, with a soft color palette and a slightly dreamy atmosphere."
 image = pipe(