return kw

check_app.py

@@ -12,13 +12,17 @@ import gradio as gr
 def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, seed=None, progress=gr.Progress()):
     generator = None
     if seed is not None:
-        generator = torch.Generator("cpu").manual_seed(seed)
+        generator = torch.Generator("cuda").manual_seed(seed)
     print("Start generating")
     # Wrapper to track progress
     def callback(pipe, step_index, timestep, callback_kwargs): # pipe, step_index, timestep, callback_kwargs
+        print(f" callback => {pipe}, {step_index}, {timestep}, {callback_kwargs} ")
+        if step_index == None:
+            step_index = 0
         cur_prg = step_index / num_steps
         print(f"Progressing {cur_prg} Step {step_index}/{num_steps}")
         progress(cur_prg, desc=f"Step {step_index}/{num_steps}")
+        return callback_kwargs
 
     if isinstance(pipe, StableDiffusion3Pipeline):
         image = pipe(