From 7273d9b89fde88a648210f65a9a2d82d9ebb959e Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 05:05:23 -0800 Subject: [PATCH] Update sd_hijack.py --- modules/sd_hijack.py | 23 ++--------------------- 1 file changed, 2 insertions(+), 21 deletions(-) diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index 6d4a8930..f5a5a696 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -136,25 +136,10 @@ class StableDiffusionModelHijack: self.embedding_db.add_embedding_dir(cmd_opts.embeddings_dir) def apply_optimizations(self, option=None): - try: - self.optimization_method = apply_optimizations(option) - except Exception as e: - errors.display(e, "applying cross attention optimization") - undo_optimizations() + pass def convert_sdxl_to_ssd(self, m): - """Converts an SDXL model to a Segmind Stable Diffusion model (see https://huggingface.co/segmind/SSD-1B)""" - - delattr(m.model.diffusion_model.middle_block, '1') - delattr(m.model.diffusion_model.middle_block, '2') - for i in ['9', '8', '7', '6', '5', '4']: - delattr(m.model.diffusion_model.input_blocks[7][1].transformer_blocks, i) - delattr(m.model.diffusion_model.input_blocks[8][1].transformer_blocks, i) - delattr(m.model.diffusion_model.output_blocks[0][1].transformer_blocks, i) - delattr(m.model.diffusion_model.output_blocks[1][1].transformer_blocks, i) - delattr(m.model.diffusion_model.output_blocks[4][1].transformer_blocks, '1') - delattr(m.model.diffusion_model.output_blocks[5][1].transformer_blocks, '1') - devices.torch_gc() + pass def hijack(self, m): conditioner = getattr(m, 'conditioner', None) @@ -199,8 +184,6 @@ class StableDiffusionModelHijack: if m.cond_stage_key == "edit": sd_hijack_unet.hijack_ddpm_edit() - self.apply_optimizations() - self.clip = m.cond_stage_model def flatten(el): @@ -223,11 +206,9 @@ class StableDiffusionModelHijack: else: sd_unet.original_forward = None - def undo_hijack(self, m): pass - def apply_circular(self, enable): if self.circular_enabled == enable: return