Hugging Face's logo

fusing
/
sd-v1-4-flax

Model card Files Community
2
valhalla commited on
Commit
da102a5
·
1 Parent(s): 0ca47f8

update weights

Browse files
Files changed (4) hide show
  1. unet/config.json +3 -5
  2. unet/diffusion_flax_model.msgpack +2 -2
  3. vae/config.json +3 -5
  4. vae/diffusion_flax_model.msgpack +2 -2
unet/config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
- "architectures": [
3
- "2D"
4
- ],
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
7
  320,
@@ -16,12 +15,11 @@
16
  "CrossAttnDownBlock2D",
17
  "DownBlock2D"
18
  ],
19
- "dropout": 0.1,
20
  "in_channels": 4,
21
  "layers_per_block": 2,
22
  "out_channels": 4,
23
  "sample_size": 64,
24
- "transformers_version": "4.21.0.dev0",
25
  "up_block_types": [
26
  "UpBlock2D",
27
  "CrossAttnUpBlock2D",
 
1
  {
2
+ "_class_name": "FlaxUNet2DConditionModel",
3
+ "_diffusers_version": "0.4.0.dev0",
 
4
  "attention_head_dim": 8,
5
  "block_out_channels": [
6
  320,
 
15
  "CrossAttnDownBlock2D",
16
  "DownBlock2D"
17
  ],
18
+ "dropout": 0.0,
19
  "in_channels": 4,
20
  "layers_per_block": 2,
21
  "out_channels": 4,
22
  "sample_size": 64,
 
23
  "up_block_types": [
24
  "UpBlock2D",
25
  "CrossAttnUpBlock2D",
unet/diffusion_flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8599453adfc763a71d6291bd92be13c61e9c970216755dc4e10b41c72cce19fb
3
- size 3438108359
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00d880e6c6bee303c1fdd0ccb3ace7549de046b15ab658bd4b3d52f1574dcfd9
3
+ size 3438108367
vae/config.json CHANGED
@@ -1,15 +1,13 @@
1
  {
 
 
2
  "act_fn": "silu",
3
- "architectures": [
4
- "encoderKL"
5
- ],
6
  "block_out_channels": [
7
  128,
8
  256,
9
  512,
10
  512
11
  ],
12
- "double_z": true,
13
  "down_block_types": [
14
  "DownEncoderBlock2D",
15
  "DownEncoderBlock2D",
@@ -19,9 +17,9 @@
19
  "in_channels": 3,
20
  "latent_channels": 4,
21
  "layers_per_block": 2,
 
22
  "out_channels": 3,
23
  "sample_size": 512,
24
- "transformers_version": "4.21.0.dev0",
25
  "up_block_types": [
26
  "UpDecoderBlock2D",
27
  "UpDecoderBlock2D",
 
1
  {
2
+ "_class_name": "FlaxAutoencoderKL",
3
+ "_diffusers_version": "0.4.0.dev0",
4
  "act_fn": "silu",
 
 
 
5
  "block_out_channels": [
6
  128,
7
  256,
8
  512,
9
  512
10
  ],
 
11
  "down_block_types": [
12
  "DownEncoderBlock2D",
13
  "DownEncoderBlock2D",
 
17
  "in_channels": 3,
18
  "latent_channels": 4,
19
  "layers_per_block": 2,
20
+ "norm_num_groups": 32,
21
  "out_channels": 3,
22
  "sample_size": 512,
 
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
25
  "UpDecoderBlock2D",
vae/diffusion_flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fabc9b1c3fc40e728272e0e1f38ac523b556489db405f5d59f035a477d9ca16b
3
- size 334623829
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed39fc57b0224dec2c0cc6f9a532633ccef89815e40b544a6b8f38f422023d8a
3
+ size 334623853