lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-26 01:09:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Intergrate Native AutoEncoderKL

Browse Source
This commit is contained in:
layerdiffusion
2024-07-31 21:08:22 -07:00
parent 2c3afff371
commit 0d079a846d
9 changed files with 473 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
ldm_patched/modules/sd.py
View File

@@ -163,10 +163,7 @@ class CLIP:
return self.patcher.get_key_patches()
class VAE:
def __init__(self, model=None, mapping=None, device=None, dtype=None, no_init=False):
if mapping is None:
mapping = {}
def __init__(self, model=None, device=None, dtype=None, no_init=False):
if no_init:
return
@@ -176,7 +173,6 @@ class VAE:
self.latent_channels = 4
self.first_stage_model = model.eval()
self.state_dict_mapping = mapping
if device is None:
device = model_management.vae_device()
@@ -202,7 +198,6 @@ class VAE:
n.downscale_ratio = self.downscale_ratio
n.latent_channels = self.latent_channels
n.first_stage_model = self.first_stage_model
n.state_dict_mapping = self.state_dict_mapping
n.device = self.device
n.vae_dtype = self.vae_dtype
n.output_device = self.output_device