998bf52
ce3d7bb
998bf52
|
|
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 |
