fix broken progress api by previous rework
This commit is contained in:
parent
17a2076f72
commit
be27fd4690
1 changed files with 3 additions and 0 deletions
|
@ -4,6 +4,7 @@ import json
|
|||
import os
|
||||
import sys
|
||||
from collections import OrderedDict
|
||||
import time
|
||||
|
||||
import gradio as gr
|
||||
import tqdm
|
||||
|
@ -132,6 +133,7 @@ class State:
|
|||
current_image = None
|
||||
current_image_sampling_step = 0
|
||||
textinfo = None
|
||||
time_start = None
|
||||
|
||||
def skip(self):
|
||||
self.skipped = True
|
||||
|
@ -168,6 +170,7 @@ class State:
|
|||
self.skipped = False
|
||||
self.interrupted = False
|
||||
self.textinfo = None
|
||||
self.time_start = time.time()
|
||||
|
||||
devices.torch_gc()
|
||||
|
||||
|
|
Loading…
Reference in a new issue