Commit be27fd46 authored by evshiron's avatar evshiron

fix broken progress api by previous rework

parent 17a2076f
...@@ -4,6 +4,7 @@ import json ...@@ -4,6 +4,7 @@ import json
import os import os
import sys import sys
from collections import OrderedDict from collections import OrderedDict
import time
import gradio as gr import gradio as gr
import tqdm import tqdm
...@@ -132,6 +133,7 @@ class State: ...@@ -132,6 +133,7 @@ class State:
current_image = None current_image = None
current_image_sampling_step = 0 current_image_sampling_step = 0
textinfo = None textinfo = None
time_start = None
def skip(self): def skip(self):
self.skipped = True self.skipped = True
...@@ -168,6 +170,7 @@ class State: ...@@ -168,6 +170,7 @@ class State:
self.skipped = False self.skipped = False
self.interrupted = False self.interrupted = False
self.textinfo = None self.textinfo = None
self.time_start = time.time()
devices.torch_gc() devices.torch_gc()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment