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