Update acestep/gradio_ui/events/results_handlers.py

acestep/gradio_ui/events/results_handlers.py

@@ -428,7 +428,6 @@ def generate_with_progress(
         use_cot_caption=use_cot_caption,
         use_cot_language=use_cot_language,
         use_constrained_decoding=True,
-        task_id=task_id,
     )
 
     if isinstance(active_seed_param, str) and active_seed_param.strip():
@@ -466,6 +465,7 @@ def generate_with_progress(
             config=gen_config,
             save_dir=None,
             progress=progress,
+            task_id=task_id,
         )
     except Exception as e:
         if e.__class__.__name__ == "TimeoutSavedException":