From af0b04cc16ce703b2f6b7a06edbb5f301a804a94 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Wed, 31 Jul 2024 21:40:18 -0700
Subject: [PATCH] store huggingface vars in VAE

---
 backend/nn/autoencoder_kl.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/backend/nn/autoencoder_kl.py b/backend/nn/autoencoder_kl.py
index 5b081f8d..0255d576 100644
--- a/backend/nn/autoencoder_kl.py
+++ b/backend/nn/autoencoder_kl.py
@@ -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)