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

store huggingface vars in VAE

Browse Source
This commit is contained in:
layerdiffusion
2024-07-31 21:40:18 -07:00
parent 3079b81547
commit af0b04cc16
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/nn/autoencoder_kl.py
View File

@@ -405,6 +405,8 @@ class IntegratedAutoencoderKL(nn.Module, ConfigMixin):
self.quant_conv = nn.Conv2d(2 * latent_channels, 2 * latent_channels, 1)
self.post_quant_conv = nn.Conv2d(latent_channels, latent_channels, 1)
self.embed_dim = latent_channels
self.scaling_factor = scaling_factor
self.shift_factor = shift_factor
def encode(self, x, regulation=None):
z = self.encoder(x)