StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add a status logger in modules.shared

Browse Source
This commit is contained in:
Aarni Koskela 2023-06-30 13:24:17 +03:00
parent e430344347
commit 522a8b9f62
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/shared.py
View File

@ -4,6 +4,7 @@ import os
import sys import sys
import threading import threading
import time import time
import logging
import gradio as gr import gradio as gr
import torch import torch
@ -18,6 +19,8 @@ from modules.paths_internal import models_path, script_path, data_path, sd_confi
from ldm.models.diffusion.ddpm import LatentDiffusion from ldm.models.diffusion.ddpm import LatentDiffusion
from typing import Optional from typing import Optional
log = logging.getLogger(__name__)
demo = None demo = None
parser = cmd_args.parser parser = cmd_args.parser
@ -144,12 +147,15 @@ class State:
def request_restart(self) -> None: def request_restart(self) -> None:
self.interrupt() self.interrupt()
self.server_command = "restart" self.server_command = "restart"
log.info("Received restart request")
def skip(self): def skip(self):
self.skipped = True self.skipped = True
log.info("Received skip request")
def interrupt(self): def interrupt(self):
self.interrupted = True self.interrupted = True
log.info("Received interrupt request")
def nextjob(self): def nextjob(self):
if opts.live_previews_enable and opts.show_progress_every_n_steps == -1: if opts.live_previews_enable and opts.show_progress_every_n_steps == -1:
@ -189,8 +195,11 @@ class State:
self.time_start = time.time() self.time_start = time.time()
self.job = job self.job = job
devices.torch_gc() devices.torch_gc()
log.info("Starting job %s", job)
def end(self): def end(self):
duration = time.time() - self.time_start
log.info("Ending job %s (%.2f seconds)", self.job, duration)
self.job = "" self.job = ""
self.job_count = 0 self.job_count = 0