Update app.py

app.py

@@ -79,7 +79,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-def scheduler_swap_callback(pipeline, step_index, timestep):
+def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
         # adjust the batch_size of prompt_embeds according to guidance_scale
     if step_index == int(pipeline.num_timesteps * 0.1):
         print("-- swapping scheduler --")