From 81dd86bf64119c31b87d0b4f583a6f9d44ccb67c Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Mon, 28 Oct 2024 15:32:01 -0400 Subject: [PATCH] update Signed-off-by: Vladimir Mandic --- repositories/ip_instruct | 1 - scripts/k_diff.py | 7 ++++++- 2 files changed, 6 insertions(+), 2 deletions(-) delete mode 160000 repositories/ip_instruct diff --git a/repositories/ip_instruct b/repositories/ip_instruct deleted file mode 160000 index 004d91f09..000000000 --- a/repositories/ip_instruct +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 004d91f09c6492133573404664c0341cbd4d19b8 diff --git a/scripts/k_diff.py b/scripts/k_diff.py index 3ce8207da..711115aa1 100644 --- a/scripts/k_diff.py +++ b/scripts/k_diff.py @@ -8,7 +8,10 @@ from modules import scripts, processing, shared, sd_models class Script(scripts.Script): supported_models = ['sd', 'sdxl'] orig_pipe = None - library = importlib.import_module('k_diffusion') + try: + library = importlib.import_module('k_diffusion') + except Exception: + library = None def title(self): return 'K-Diffusion' @@ -40,6 +43,8 @@ class Script(scripts.Script): if shared.sd_model_type not in self.supported_models: shared.log.warning(f'K-Diffusion: class={shared.sd_model.__class__.__name__} model={shared.sd_model_type} required={self.supported_models}') return None + if self.library is None: + return cls = None if shared.sd_model_type == "sd": cls = diffusers.pipelines.StableDiffusionKDiffusionPipeline