remove dirty cond_stage_key

This commit is contained in:
layerdiffusion
2024-08-05 11:39:20 -07:00
parent d77582aa5a
commit 27641043ee
3 changed files with 2 additions and 3 deletions

View File

@@ -34,7 +34,6 @@ class ForgeDiffusionEngine:
self.first_stage_model = None # set this so that you can change VAE in UI
# WebUI Dirty Legacy
self.cond_stage_key = 'txt'
self.is_sd3 = False
self.latent_channels = 4
self.is_sdxl = False

View File

@@ -1579,7 +1579,7 @@ class StableDiffusionProcessingImg2Img(StableDiffusionProcessing):
def init(self, all_prompts, all_seeds, all_subseeds):
self.extra_generation_params["Denoising strength"] = self.denoising_strength
self.image_cfg_scale: float = self.image_cfg_scale if shared.sd_model.cond_stage_key == "edit" else None
self.image_cfg_scale: float = None
self.sampler = sd_samplers.create_sampler(self.sampler_name, self.sd_model)
crop_region = None

View File

@@ -938,7 +938,7 @@ def create_ui():
settings.add_functionality(demo)
update_image_cfg_scale_visibility = lambda: gr.update(visible=shared.sd_model and shared.sd_model.cond_stage_key == "edit")
update_image_cfg_scale_visibility = lambda: gr.update(visible=False)
settings.text_settings.change(fn=update_image_cfg_scale_visibility, inputs=[], outputs=[image_cfg_scale])
demo.load(fn=update_image_cfg_scale_visibility, inputs=[], outputs=[image_cfg_scale])