Error in infer_method_patch.py

[Goerf]

This happened on a Mac (26.5.1) in Safari:

File "/Users/Hans/ACE-Step-1.5/acestep/ui/aceflow/app.py", line 3211, in _run_job
raise RuntimeError (result .error or result .status_message or "Errore sconosciuto")
│ │ │ └ 'Error: ServiceGenerateExecuteMixin._execute_service_generate_diffusion() got an unexpected keyword argument 'retake_seed'...
│ │ └ GenerationResult(audios=[], status_message='Error: ServiceGenerateExecuteMixin._execute_service_generate_diffusion() got an u...
│ └ "ServiceGenerateExecuteMixin._execute_service_generate_diffusion() got an unexpected keyword argument 'retake_seed'"
└ GenerationResult(audios=[], status_message='Error: ServiceGenerateExecuteMixin._execute_service_generate_diffusion() got an u...

RuntimeError: ServiceGenerateExecuteMixin._execute_service_generate_diffusion() got an unexpected keyword argument 'retake_seed'

Given this to Claude how analyzed it an proposed a solution which seems to work:

The bug: in acestep/ui/aceflow/infer_method_patch.py, the monkey-patched _execute_service_generate_diffusion wrapper had a stale signature missing retake_seed, retake_variance, and flow_edit_ctx, which service_generate.py now passes as kwargs. Fixed by adding those params to the wrapper and forwarding them to original_fn.

I've uploaded the patched file, can you check if this is the correct solution for the bug:

infer_method_patch.py