Spaces:
Running
on
Zero
Running
on
Zero
fixed has
Browse files- check_app.py +2 -2
check_app.py
CHANGED
@@ -147,6 +147,8 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
|
|
147 |
num_inference_steps=num_steps,
|
148 |
guidance_scale=guidance_scale,
|
149 |
generator=generator,
|
|
|
|
|
150 |
).images[0]
|
151 |
elif has_callback_on_step_end and not has_guidance_scale:
|
152 |
print("has callback_on_step_end and NO guidance_scale")
|
@@ -162,8 +164,6 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
|
|
162 |
prompt,
|
163 |
num_inference_steps=num_steps,
|
164 |
generator=generator,
|
165 |
-
callback=callback,
|
166 |
-
callback_steps=1,
|
167 |
).images[0]
|
168 |
|
169 |
return image
|
|
|
147 |
num_inference_steps=num_steps,
|
148 |
guidance_scale=guidance_scale,
|
149 |
generator=generator,
|
150 |
+
callback=callback,
|
151 |
+
callback_steps=1,
|
152 |
).images[0]
|
153 |
elif has_callback_on_step_end and not has_guidance_scale:
|
154 |
print("has callback_on_step_end and NO guidance_scale")
|
|
|
164 |
prompt,
|
165 |
num_inference_steps=num_steps,
|
166 |
generator=generator,
|
|
|
|
|
167 |
).images[0]
|
168 |
|
169 |
return image
|