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
|