Add UNet and VAE checkpoints
Browse files- unet/optimizer.bin +3 -0
- unet/random_states_0.pkl +3 -0
- unet/scaler.pt +3 -0
- unet/scheduler.bin +3 -0
- unet/self_attn_clip/pytorch_model.bin +3 -0
- unet/unet/config.json +8 -0
- unet/unet/diffusion_pytorch_model.safetensors +3 -0
- unet/unet_ema/config.json +15 -0
- unet/unet_ema/diffusion_pytorch_model.safetensors +3 -0
- vae/discriminator/pytorch_model.bin +3 -0
- vae/ema_vae/config.json +45 -0
- vae/ema_vae/diffusion_pytorch_model.safetensors +3 -0
- vae/optimizer.bin +3 -0
- vae/optimizer_1.bin +3 -0
- vae/random_states_0.pkl +3 -0
- vae/scheduler.bin +3 -0
- vae/scheduler_1.bin +3 -0
- vae/vae/config.json +38 -0
- vae/vae/diffusion_pytorch_model.safetensors +3 -0
unet/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a38956d8a345352a950863efc1ec71e4b2b0ca462ddfd655a9fccf81113cb618
|
| 3 |
+
size 7098849886
|
unet/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1d994dda485d3c6702937de2c15ae78d39028dbfeaf18c3c1fc6b4ea23d8fcdf
|
| 3 |
+
size 14408
|
unet/scaler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3bdae41e930d90d71b0f1e255ed6417b90c951bddcd99b75d9e09f967243eea1
|
| 3 |
+
size 988
|
unet/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5f3b950aefa79e94b1b89bbdbaf4ed43f10512c188a34f46c2b33dbd8e7b85bf
|
| 3 |
+
size 1064
|
unet/self_attn_clip/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:42e63db1940f8e02c1a44287f87516d94c2cdc398a7ab50ecba00de1772de61a
|
| 3 |
+
size 9451712
|
unet/unet/config.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "FiLMUNet2DConditionModel",
|
| 3 |
+
"_diffusers_version": "0.32.2",
|
| 4 |
+
"embed_dim_clip": 768,
|
| 5 |
+
"embed_dim_plucker": 78,
|
| 6 |
+
"in_channels": 9,
|
| 7 |
+
"pretrained_model_name_or_path": "stabilityai/stable-diffusion-2"
|
| 8 |
+
}
|
unet/unet/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2ac730fba6ab3b24d11822bf10cb640af6bc05d637fe82b699440a80447bacc1
|
| 3 |
+
size 3539754920
|
unet/unet_ema/config.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "FiLMUNet2DConditionModel",
|
| 3 |
+
"_diffusers_version": "0.32.2",
|
| 4 |
+
"decay": 0.9999,
|
| 5 |
+
"embed_dim_clip": 768,
|
| 6 |
+
"embed_dim_plucker": 78,
|
| 7 |
+
"in_channels": 9,
|
| 8 |
+
"inv_gamma": 1.0,
|
| 9 |
+
"min_decay": 0.0,
|
| 10 |
+
"optimization_step": 67000,
|
| 11 |
+
"power": 0.6666666666666666,
|
| 12 |
+
"pretrained_model_name_or_path": "stabilityai/stable-diffusion-2",
|
| 13 |
+
"update_after_step": 0,
|
| 14 |
+
"use_ema_warmup": false
|
| 15 |
+
}
|
unet/unet_ema/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4e73c2232204cad327a2b88aa6a4245712ca9e6ed4bcd05bb18d005c20e7fe6c
|
| 3 |
+
size 3539754920
|
vae/discriminator/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:84e980d8431d228ad9ab24808032c00d89644a3eeca0375e02ec385fd49a9cd5
|
| 3 |
+
size 11081328
|
vae/ema_vae/config.json
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "AutoencoderKL",
|
| 3 |
+
"_diffusers_version": "0.30.3",
|
| 4 |
+
"_name_or_path": "Intel/ldm3d-4c",
|
| 5 |
+
"act_fn": "silu",
|
| 6 |
+
"block_out_channels": [
|
| 7 |
+
128,
|
| 8 |
+
256,
|
| 9 |
+
512,
|
| 10 |
+
512
|
| 11 |
+
],
|
| 12 |
+
"decay": 0.9999,
|
| 13 |
+
"down_block_types": [
|
| 14 |
+
"DownEncoderBlock2D",
|
| 15 |
+
"DownEncoderBlock2D",
|
| 16 |
+
"DownEncoderBlock2D",
|
| 17 |
+
"DownEncoderBlock2D"
|
| 18 |
+
],
|
| 19 |
+
"force_upcast": true,
|
| 20 |
+
"in_channels": 4,
|
| 21 |
+
"inv_gamma": 1.0,
|
| 22 |
+
"latent_channels": 4,
|
| 23 |
+
"latents_mean": null,
|
| 24 |
+
"latents_std": null,
|
| 25 |
+
"layers_per_block": 2,
|
| 26 |
+
"mid_block_add_attention": true,
|
| 27 |
+
"min_decay": 0.0,
|
| 28 |
+
"norm_num_groups": 32,
|
| 29 |
+
"optimization_step": 6000,
|
| 30 |
+
"out_channels": 4,
|
| 31 |
+
"power": 0.6666666666666666,
|
| 32 |
+
"sample_size": 256,
|
| 33 |
+
"scaling_factor": 0.18215,
|
| 34 |
+
"shift_factor": null,
|
| 35 |
+
"up_block_types": [
|
| 36 |
+
"UpDecoderBlock2D",
|
| 37 |
+
"UpDecoderBlock2D",
|
| 38 |
+
"UpDecoderBlock2D",
|
| 39 |
+
"UpDecoderBlock2D"
|
| 40 |
+
],
|
| 41 |
+
"update_after_step": 0,
|
| 42 |
+
"use_ema_warmup": false,
|
| 43 |
+
"use_post_quant_conv": true,
|
| 44 |
+
"use_quant_conv": true
|
| 45 |
+
}
|
vae/ema_vae/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:333ef0a8c1c2964c6ca09746da2cda96f9d6903f87b9afe89fc6649b56788d7d
|
| 3 |
+
size 334652496
|
vae/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:014c1c42aa3ba39f35ca38613565e492073a32422aba11f715cc166132b7247a
|
| 3 |
+
size 669457018
|
vae/optimizer_1.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:10be8077acb2e3152a9e4926eee4840f94a0f56911a2d52b645f8d060648ed3a
|
| 3 |
+
size 22144717
|
vae/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:27a70049082c5a1f1df99994748c0bcaf33bb730fcf038b3561f245aedecbad6
|
| 3 |
+
size 14344
|
vae/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a7fff6aa20c4123f26aa862c894f7dac7a89b1494a8aab0a5fba5f3e1d4ab0b9
|
| 3 |
+
size 1000
|
vae/scheduler_1.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a24de919d78a209a8db5f3b903b4b21e406148601affdc758fc2f65782a6244d
|
| 3 |
+
size 1008
|
vae/vae/config.json
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "AutoencoderKL",
|
| 3 |
+
"_diffusers_version": "0.30.3",
|
| 4 |
+
"_name_or_path": "Intel/ldm3d-4c",
|
| 5 |
+
"act_fn": "silu",
|
| 6 |
+
"block_out_channels": [
|
| 7 |
+
128,
|
| 8 |
+
256,
|
| 9 |
+
512,
|
| 10 |
+
512
|
| 11 |
+
],
|
| 12 |
+
"down_block_types": [
|
| 13 |
+
"DownEncoderBlock2D",
|
| 14 |
+
"DownEncoderBlock2D",
|
| 15 |
+
"DownEncoderBlock2D",
|
| 16 |
+
"DownEncoderBlock2D"
|
| 17 |
+
],
|
| 18 |
+
"force_upcast": true,
|
| 19 |
+
"in_channels": 4,
|
| 20 |
+
"latent_channels": 4,
|
| 21 |
+
"latents_mean": null,
|
| 22 |
+
"latents_std": null,
|
| 23 |
+
"layers_per_block": 2,
|
| 24 |
+
"mid_block_add_attention": true,
|
| 25 |
+
"norm_num_groups": 32,
|
| 26 |
+
"out_channels": 4,
|
| 27 |
+
"sample_size": 256,
|
| 28 |
+
"scaling_factor": 0.18215,
|
| 29 |
+
"shift_factor": null,
|
| 30 |
+
"up_block_types": [
|
| 31 |
+
"UpDecoderBlock2D",
|
| 32 |
+
"UpDecoderBlock2D",
|
| 33 |
+
"UpDecoderBlock2D",
|
| 34 |
+
"UpDecoderBlock2D"
|
| 35 |
+
],
|
| 36 |
+
"use_post_quant_conv": true,
|
| 37 |
+
"use_quant_conv": true
|
| 38 |
+
}
|
vae/vae/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:eae1a0654aacddccd99bb903eb97e5b375b8b59cbb279352ad535d3d83a8e8e2
|
| 3 |
+
size 334652496
|