| from gradio import Progress | |
| def create_progress_updater(start: int, total: int, desc: str, progress: Progress): | |
| def updater(pipe, step, timestep, callback_kwargs): | |
| if progress: | |
| progress((step + start + 1, total), desc=desc) | |
| return callback_kwargs | |
| return updater |