Fix syntax error and hardcoded device

README.md

@@ -57,7 +57,7 @@ pipe = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
 positive_magic = {
-    "en": "Ultra HD, 4K, cinematic composition." # for english prompt,
+    "en": "Ultra HD, 4K, cinematic composition.", # for english prompt,
     "zh": "超清，4K，电影级构图" # for chinese prompt,
 }
 
@@ -87,7 +87,7 @@ image = pipe(
     height=height,
     num_inference_steps=50,
     true_cfg_scale=4.0,
-    generator=torch.Generator(device="cuda").manual_seed(42)
+    generator=torch.Generator(device=device).manual_seed(42)
 ).images[0]
 
 image.save("example.png")