model weights from paper added

app.py

@@ -48,7 +48,7 @@ class AttributionModel:
         self.vae = AutoencoderKL.from_pretrained(
             'stabilityai/stable-diffusion-2', subfolder="vae"
         )
-        self.vae = customize_vae_decoder(self.vae, 128, "qkv", "all", False, 1.0)
+        self.vae = customize_vae_decoder(self.vae, 128, "deqkv", "all", False, 1.0)
 
         self.mapping_network = MappingNetwork(32, 0, 128, None, num_layers=2, w_avg_beta=None, normalization = False)
         

checkpoints/decoding_network.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:161e00989c8a1e1461087d4cafbf65924d526bf67f3376a798104fd174e01d93
-size 94610003
+oid sha256:8c127e07ca3746a35ede928e6fa8cc9a9a7499ab972c13abd417e2f2bbe38dfa
+size 94614483

checkpoints/mapping_network.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0a228b227adfb154ff1afef38de04814c7e35d77e295417b40a85f9f4ff266d2
-size 84623
+oid sha256:9f9c3902af67da17ef1d65561871a0a9aee2ac9521f048785f3bfa09b8946626
+size 84687

checkpoints/vae_decoder.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:32824847ca5b95304b5d491501cef2ee9c9ea4409ec0d4567732cf7942b15087
-size 198804905
+oid sha256:aef96ce90432b1b66c930804d011ff508fcfdc43eba6256987dbae7a094b475a
+size 204440985