Spaces:

multimodalart
/

stable-cascade

Running on Zero

multimodalart HF staff commited on Mar 14

Commit

ff1a3bd

•

1 Parent(s): f22f594

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -98,8 +98,7 @@ def generate(
         guidance_scale=prior_guidance_scale,
         num_images_per_prompt=num_images_per_prompt,
         generator=generator,
-        callback=callback_prior,
-        callback_steps=callback_steps
     )
     if PREVIEW_IMAGES:

         guidance_scale=prior_guidance_scale,
         num_images_per_prompt=num_images_per_prompt,
         generator=generator,
+        callback_on_step_end=callback_prior,
     )
     if PREVIEW_IMAGES: