From 5cfad82bea5c719fe3ca960e7ba79ca0f23ecb8a Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 08:13:17 -0800 Subject: [PATCH] i --- modules/sd_samplers_common.py | 1 - modules_forge/forge_loader.py | 7 +++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 73335083..63fb1ba3 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -92,7 +92,6 @@ def images_tensor_to_samples(image, approximation=None, model=None): else: if model is None: model = shared.sd_model - model.first_stage_model.to(devices.dtype_vae) image = image.to(shared.device, dtype=devices.dtype_vae) image = image * 2 - 1 diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 8f68ef51..07891db1 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -225,7 +225,14 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): sample = forge_object.unet.model.model_config.latent_format.process_out(sample) return forge_object.vae.decode(sample).movedim(-1, 1) * 2.0 - 1.0 + def patched_encode_first_stage(sample): + sample = forge_object.vae.encode(sample) + sample = forge_object.unet.model.model_config.latent_format.process_in(sample) + return sample + + sd_model.encode_first_stage = lambda x: x sd_model.decode_first_stage = patched_decode_first_stage + sd_model.encode_first_stage = patched_encode_first_stage sd_model.unet_patcher = forge_object.unet sd_model.clip_patcher = forge_object.clip.patcher