patrickvonplaten commited on
Commit
4e2e5b3
1 Parent(s): 8abea1f

Update README.md

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