Hugging Face's logo

ribster
/
sd-yff-jax

Model card Files Community
root commited on
Commit
375a5bd
•
1 Parent(s): 928c0c2

v0.2

Browse files
Files changed (5) hide show
  1. model_index.json +2 -2
  2. unet/diffusion_flax_model.msgpack +2 -2
  3. vae/config.jsontemp +29 -0
  4. vae/diffusion_flax_model.msgpack +2 -2
  5. vae/diffusion_flax_modeltemp.msgpack +3 -0
model_index.json CHANGED
@@ -6,8 +6,8 @@
6
  "CLIPImageProcessor"
7
  ],
8
  "safety_checker": [
9
- "stable_diffusion",
10
- "FlaxStableDiffusionSafetyChecker"
11
  ],
12
  "scheduler": [
13
  "diffusers",
 
6
  "CLIPImageProcessor"
7
  ],
8
  "safety_checker": [
9
+ null,
10
+ null
11
  ],
12
  "scheduler": [
13
  "diffusers",
unet/diffusion_flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bcdfef4bd4f7032d8a4bf4e450c58d14a17d84613130e5e796b676360e98f613
3
- size 1719067125
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fd737bf39225743fb554a4b48b8c5793a75225d6968dce0f9f545c95931ade6
3
+ size 3438108367
vae/config.jsontemp ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "FlaxAutoencoderKL",
3
+ "_diffusers_version": "0.12.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",
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D"
16
+ ],
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",
26
+ "UpDecoderBlock2D",
27
+ "UpDecoderBlock2D"
28
+ ]
29
+ }
vae/diffusion_flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0e3e81680c6669960f086eca278c8a1f89dd2ed7eadef806e049b56e426c1809
3
- size 167316369
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed39fc57b0224dec2c0cc6f9a532633ccef89815e40b544a6b8f38f422023d8a
3
+ size 334623853
vae/diffusion_flax_modeltemp.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ede690dbf32f7b28d813fea63ff821c14df279cc0fe4b89502d052ee60e0d3a5
3
+ size 109552740