Maitreyapatel commited on
Commit
537c1bb
1 Parent(s): b96d5dc

model weights from paper added

Browse files
app.py CHANGED
@@ -48,7 +48,7 @@ class AttributionModel:
48
  self.vae = AutoencoderKL.from_pretrained(
49
  'stabilityai/stable-diffusion-2', subfolder="vae"
50
  )
51
- self.vae = customize_vae_decoder(self.vae, 128, "qkv", "all", False, 1.0)
52
 
53
  self.mapping_network = MappingNetwork(32, 0, 128, None, num_layers=2, w_avg_beta=None, normalization = False)
54
 
 
48
  self.vae = AutoencoderKL.from_pretrained(
49
  'stabilityai/stable-diffusion-2', subfolder="vae"
50
  )
51
+ self.vae = customize_vae_decoder(self.vae, 128, "deqkv", "all", False, 1.0)
52
 
53
  self.mapping_network = MappingNetwork(32, 0, 128, None, num_layers=2, w_avg_beta=None, normalization = False)
54
 
checkpoints/decoding_network.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:161e00989c8a1e1461087d4cafbf65924d526bf67f3376a798104fd174e01d93
3
- size 94610003
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c127e07ca3746a35ede928e6fa8cc9a9a7499ab972c13abd417e2f2bbe38dfa
3
+ size 94614483
checkpoints/mapping_network.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0a228b227adfb154ff1afef38de04814c7e35d77e295417b40a85f9f4ff266d2
3
- size 84623
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f9c3902af67da17ef1d65561871a0a9aee2ac9521f048785f3bfa09b8946626
3
+ size 84687
checkpoints/vae_decoder.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:32824847ca5b95304b5d491501cef2ee9c9ea4409ec0d4567732cf7942b15087
3
- size 198804905
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aef96ce90432b1b66c930804d011ff508fcfdc43eba6256987dbae7a094b475a
3
+ size 204440985