LPX55 commited on
Commit
e5991c5
·
verified ·
1 Parent(s): bda1878

Update app_v4.py

Browse files
Files changed (1) hide show
  1. app_v4.py +2 -2
app_v4.py CHANGED
@@ -162,7 +162,7 @@ def generate_focus(control_image, focus_list):
162
 
163
  @spaces.GPU(duration=12, progress=gr.Progress(track_tqdm=True))
164
  @torch.no_grad()
165
- def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale, seed, guidance_end):
166
  generator = torch.Generator().manual_seed(seed)
167
  # Load control image
168
  control_image = load_image(control_image)
@@ -186,7 +186,7 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
186
  guidance_scale=guidance_scale,
187
  height=control_image.size[1],
188
  width=control_image.size[0],
189
- control_guidance_start=0.0,
190
  control_guidance_end=guidance_end,
191
  ).images[0]
192
  # print("Type: " + str(type(image)))
 
162
 
163
  @spaces.GPU(duration=12, progress=gr.Progress(track_tqdm=True))
164
  @torch.no_grad()
165
+ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale, seed, guidance_start, guidance_end):
166
  generator = torch.Generator().manual_seed(seed)
167
  # Load control image
168
  control_image = load_image(control_image)
 
186
  guidance_scale=guidance_scale,
187
  height=control_image.size[1],
188
  width=control_image.size[0],
189
+ control_guidance_start=guidance_start,
190
  control_guidance_end=guidance_end,
191
  ).images[0]
192
  # print("Type: " + str(type(image)))