Disable loading progress bar

app.py

@@ -12,8 +12,12 @@ from lib import Config, async_call, download_civit_file, download_repo_files, ge
 
 filterwarnings("ignore", category=FutureWarning, module="diffusers")
 filterwarnings("ignore", category=FutureWarning, module="transformers")
+
 diffusers_logging.set_verbosity_error()
+diffusers_logging.disable_progress_bar()
+
 transformers_logging.set_verbosity_error()
+transformers_logging.disable_progress_bar()
 
 # the CSS `content` attribute expects a string so we need to wrap the number in quotes
 refresh_seed_js = """

lib/inference.py

@@ -169,6 +169,8 @@ def generate(
 
     start = time.perf_counter()
     log = Logger("generate")
+    log.info(f"Generating {num_images} image{'s' if num_images > 1 else ''}")
+
     loader = Loader()
     loader.load(
         KIND,
@@ -314,7 +316,7 @@ def generate(
             CURRENT_IMAGE += 1
 
     diff = time.perf_counter() - start
-    msg = f"Generated {len(images)} image{'s' if len(images) > 1 else ''} in {diff:.2f}s"
+    msg = f"Generating {len(images)} image{'s' if len(images) > 1 else ''} done in {diff:.2f}s"
     log.info(msg)
     if Info:
         Info(msg)

lib/loader.py

@@ -1,4 +1,5 @@
 import gc
+import time
 from threading import Lock
 
 import torch
@@ -145,6 +146,7 @@ class Loader:
         pipeline = Config.PIPELINES[kind]
         if self.pipe is None:
             try:
+                start = time.perf_counter()
                 self.log.info(f"Loading {model}")
                 self.model = model
                 if model.lower() in Config.MODEL_CHECKPOINTS.keys():
@@ -154,6 +156,8 @@ class Loader:
                     ).to("cuda")
                 else:
                     self.pipe = pipeline.from_pretrained(model, **kwargs).to("cuda")
+                diff = time.perf_counter() - start
+                self.log.info(f"Loading {model} done in {diff:.2f}s")
             except Exception as e:
                 self.log.error(f"Error loading {model}: {e}")
                 self.model = None