patrickvonplaten commited on
Commit
19fc41b
1 Parent(s): 6cf1aef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -66,12 +66,12 @@ model_id = "CompVis/stable-diffusion-v1-2"
66
  device = "cuda"
67
 
68
 
69
- pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
70
  pipe = pipe.to(device)
71
 
72
  prompt = "a photo of an astronaut riding a horse on mars"
73
  with autocast("cuda"):
74
- image = pipe(prompt, guidance_scale=7.5)["sample"][0]
75
 
76
  image.save("astronaut_rides_horse.png")
77
  ```
 
66
  device = "cuda"
67
 
68
 
69
+ pipe = StableDiffusionPipeline.from_pretrained(model_id)
70
  pipe = pipe.to(device)
71
 
72
  prompt = "a photo of an astronaut riding a horse on mars"
73
  with autocast("cuda"):
74
+ image = pipe(prompt)["sample"][0]
75
 
76
  image.save("astronaut_rides_horse.png")
77
  ```