fix: progress updater
Browse files
utils.py
CHANGED
@@ -2,7 +2,6 @@ from gradio import Progress
|
|
2 |
|
3 |
def create_progress_updater(start: int, total: int, desc: str, progress: Progress):
|
4 |
def updater(pipe, step, timestep, callback_kwargs):
|
5 |
-
|
6 |
-
progress((step + start + 1, total), desc=desc)
|
7 |
return callback_kwargs
|
8 |
return updater
|
|
|
2 |
|
3 |
def create_progress_updater(start: int, total: int, desc: str, progress: Progress):
|
4 |
def updater(pipe, step, timestep, callback_kwargs):
|
5 |
+
progress((step + start, total), desc=desc)
|
|
|
6 |
return callback_kwargs
|
7 |
return updater
|