Update app.py

app.py

@@ -114,6 +114,7 @@ def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
      #   pipe.vae = vae_a
      #   pipe.unet = unet_a
         torch.backends.cudnn.deterministic = False
+        pipe.unet.set_default_attn_processor()
         print("-- swapping scheduler --")
       #  pipeline.scheduler = heun_scheduler
         #pipe.scheduler.set_timesteps(num_inference_steps*.70)