Spaces:
Running
on
Zero
Running
on
Zero
Update app_v4.py
Browse files
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=
|
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)))
|