Update README.md

README.md

@@ -44,7 +44,7 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline
 
 model_id = "CompVis/stable-diffusion-v1-2-diffusers"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)  
+pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True).to("cuda")
 
 prompt = "a photograph of an astronaut riding a horse"
 with autocast("cuda"):
@@ -61,7 +61,7 @@ from diffusers import StableDiffusionPipeline, LMSDiscreteScheduler
 model_id = "CompVis/stable-diffusion-v1-2-diffusers"
 # Use the K-LMS scheduler here instead
 scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, use_auth_token=True)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, use_auth_token=True).to("cuda")
 ```