patrickvonplaten commited on
Commit
bb237df
1 Parent(s): 14477e8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -117,7 +117,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
117
  pipe.enable_model_cpu_offload()
118
 
119
  generator = torch.manual_seed(0)
120
- image = pipe(prompt, num_inference_steps=30, generator=generator, image=image).images[0]
121
 
122
  image.save('images/image_out.png')
123
  ```
 
117
  pipe.enable_model_cpu_offload()
118
 
119
  generator = torch.manual_seed(0)
120
+ image = pipe(prompt, num_inference_steps=30, generator=generator, image=control_image).images[0]
121
 
122
  image.save('images/image_out.png')
123
  ```