Commit be27fd46 authored by evshiron's avatar evshiron

fix broken progress api by previous rework

parent 17a2076f
......@@ -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()
......
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