valhalla commited on
Commit
e5cfb9a
1 Parent(s): 9852c31

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -62,7 +62,7 @@ pipe = pipe.to(device)
62
 
63
  prompt = "a photograph of an astronaut riding a horse"
64
  with autocast("cuda"):
65
- image = pipe(prompt)["sample"][0] # image here is in PIL format
66
 
67
  image.save(f"astronaut_rides_horse.png")
68
  ```
 
62
 
63
  prompt = "a photograph of an astronaut riding a horse"
64
  with autocast("cuda"):
65
+ image = pipe(prompt, generator=generator)["sample"][0] # image here is in PIL format
66
 
67
  image.save(f"astronaut_rides_horse.png")
68
  ```