RageshAntony commited on
Commit
48b54d6
·
verified ·
1 Parent(s): 0de6d5a
Files changed (1) hide show
  1. 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