mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-12-15 07:20:31 +08:00
fix broken progress api by previous rework
This commit is contained in:
parent
17a2076f72
commit
be27fd4690
@ -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 New Issue
Block a user