From e080fd1e08a676d412108bb2ecf27e559c0bfc70 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 30 Jul 2023 19:18:05 -0400 Subject: [PATCH] blackify --- invokeai/backend/model_management/models/base.py | 2 +- invokeai/backend/model_management/models/stable_diffusion.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/invokeai/backend/model_management/models/base.py b/invokeai/backend/model_management/models/base.py index 1219d4277d..d335b645c8 100644 --- a/invokeai/backend/model_management/models/base.py +++ b/invokeai/backend/model_management/models/base.py @@ -292,7 +292,7 @@ class DiffusersModel(ModelBase): ) break except Exception as e: - if not str(e).startswith('Error no file'): + if not str(e).startswith("Error no file"): print("====ERR LOAD====") print(f"{variant}: {e}") pass diff --git a/invokeai/backend/model_management/models/stable_diffusion.py b/invokeai/backend/model_management/models/stable_diffusion.py index 9e0c130e6a..a112e8bc96 100644 --- a/invokeai/backend/model_management/models/stable_diffusion.py +++ b/invokeai/backend/model_management/models/stable_diffusion.py @@ -22,6 +22,7 @@ import invokeai.backend.util.logging as logger from invokeai.app.services.config import InvokeAIAppConfig from omegaconf import OmegaConf + class StableDiffusion1ModelFormat(str, Enum): Checkpoint = "checkpoint" Diffusers = "diffusers" @@ -264,7 +265,7 @@ def _convert_ckpt_and_cache( config_file = app_config.root_path / model_config.config output_path = Path(output_path) variant = model_config.variant - pipeline_class = StableDiffusionInpaintPipeline if variant=='inpaint' else StableDiffusionPipeline + pipeline_class = StableDiffusionInpaintPipeline if variant == "inpaint" else StableDiffusionPipeline # return cached version if it exists if output_path.exists():