Epoch 20
Browse files- checkpoint-2000/optimizer.bin +3 -0
- checkpoint-2000/random_states_0.pkl +3 -0
- checkpoint-2000/scheduler.bin +3 -0
- checkpoint-2000/unet/config.json +48 -0
- checkpoint-2000/unet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-2000/unet_ema/config.json +55 -0
- checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-4000/optimizer.bin +3 -0
- checkpoint-4000/random_states_0.pkl +3 -0
- checkpoint-4000/scheduler.bin +3 -0
- checkpoint-4000/unet/config.json +48 -0
- checkpoint-4000/unet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-4000/unet_ema/config.json +55 -0
- checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors +3 -0
- unet/diffusion_pytorch_model.safetensors +1 -1
checkpoint-2000/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:881be918c2bf038a9ac2b437d656f6bcb39ddcd4766737e0acee952dcda1e37e
|
| 3 |
+
size 909800163
|
checkpoint-2000/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0a3f34f474d5cf9f26c4c96da777a7893829ac95aadc277d2f37e8149ecee76f
|
| 3 |
+
size 14663
|
checkpoint-2000/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:09e6134c4ac3863c4f5f31314057b94d331efe86f43c58118beadd59f63e8b47
|
| 3 |
+
size 563
|
checkpoint-2000/unet/config.json
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "UNet2DModel",
|
| 3 |
+
"_diffusers_version": "0.21.0.dev0",
|
| 4 |
+
"act_fn": "silu",
|
| 5 |
+
"add_attention": true,
|
| 6 |
+
"attention_head_dim": 8,
|
| 7 |
+
"block_out_channels": [
|
| 8 |
+
128,
|
| 9 |
+
128,
|
| 10 |
+
256,
|
| 11 |
+
256,
|
| 12 |
+
512,
|
| 13 |
+
512
|
| 14 |
+
],
|
| 15 |
+
"center_input_sample": false,
|
| 16 |
+
"class_embed_type": null,
|
| 17 |
+
"down_block_types": [
|
| 18 |
+
"DownBlock2D",
|
| 19 |
+
"DownBlock2D",
|
| 20 |
+
"DownBlock2D",
|
| 21 |
+
"DownBlock2D",
|
| 22 |
+
"AttnDownBlock2D",
|
| 23 |
+
"DownBlock2D"
|
| 24 |
+
],
|
| 25 |
+
"downsample_padding": 1,
|
| 26 |
+
"downsample_type": "conv",
|
| 27 |
+
"flip_sin_to_cos": true,
|
| 28 |
+
"freq_shift": 0,
|
| 29 |
+
"in_channels": 5,
|
| 30 |
+
"layers_per_block": 2,
|
| 31 |
+
"mid_block_scale_factor": 1,
|
| 32 |
+
"norm_eps": 1e-05,
|
| 33 |
+
"norm_num_groups": 32,
|
| 34 |
+
"num_class_embeds": null,
|
| 35 |
+
"out_channels": 5,
|
| 36 |
+
"resnet_time_scale_shift": "default",
|
| 37 |
+
"sample_size": 32,
|
| 38 |
+
"time_embedding_type": "positional",
|
| 39 |
+
"up_block_types": [
|
| 40 |
+
"UpBlock2D",
|
| 41 |
+
"AttnUpBlock2D",
|
| 42 |
+
"UpBlock2D",
|
| 43 |
+
"UpBlock2D",
|
| 44 |
+
"UpBlock2D",
|
| 45 |
+
"UpBlock2D"
|
| 46 |
+
],
|
| 47 |
+
"upsample_type": "conv"
|
| 48 |
+
}
|
checkpoint-2000/unet/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:195a977aa69b6c4429c0fb3a10c4c3f50dc6d1a032ddf58abb64129c75412528
|
| 3 |
+
size 454759548
|
checkpoint-2000/unet_ema/config.json
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "UNet2DModel",
|
| 3 |
+
"_diffusers_version": "0.21.0.dev0",
|
| 4 |
+
"act_fn": "silu",
|
| 5 |
+
"add_attention": true,
|
| 6 |
+
"attention_head_dim": 8,
|
| 7 |
+
"block_out_channels": [
|
| 8 |
+
128,
|
| 9 |
+
128,
|
| 10 |
+
256,
|
| 11 |
+
256,
|
| 12 |
+
512,
|
| 13 |
+
512
|
| 14 |
+
],
|
| 15 |
+
"center_input_sample": false,
|
| 16 |
+
"class_embed_type": null,
|
| 17 |
+
"decay": 0.9999,
|
| 18 |
+
"down_block_types": [
|
| 19 |
+
"DownBlock2D",
|
| 20 |
+
"DownBlock2D",
|
| 21 |
+
"DownBlock2D",
|
| 22 |
+
"DownBlock2D",
|
| 23 |
+
"AttnDownBlock2D",
|
| 24 |
+
"DownBlock2D"
|
| 25 |
+
],
|
| 26 |
+
"downsample_padding": 1,
|
| 27 |
+
"downsample_type": "conv",
|
| 28 |
+
"flip_sin_to_cos": true,
|
| 29 |
+
"freq_shift": 0,
|
| 30 |
+
"in_channels": 5,
|
| 31 |
+
"inv_gamma": 1.0,
|
| 32 |
+
"layers_per_block": 2,
|
| 33 |
+
"mid_block_scale_factor": 1,
|
| 34 |
+
"min_decay": 0.0,
|
| 35 |
+
"norm_eps": 1e-05,
|
| 36 |
+
"norm_num_groups": 32,
|
| 37 |
+
"num_class_embeds": null,
|
| 38 |
+
"optimization_step": 2000,
|
| 39 |
+
"out_channels": 5,
|
| 40 |
+
"power": 0.75,
|
| 41 |
+
"resnet_time_scale_shift": "default",
|
| 42 |
+
"sample_size": 32,
|
| 43 |
+
"time_embedding_type": "positional",
|
| 44 |
+
"up_block_types": [
|
| 45 |
+
"UpBlock2D",
|
| 46 |
+
"AttnUpBlock2D",
|
| 47 |
+
"UpBlock2D",
|
| 48 |
+
"UpBlock2D",
|
| 49 |
+
"UpBlock2D",
|
| 50 |
+
"UpBlock2D"
|
| 51 |
+
],
|
| 52 |
+
"update_after_step": 0,
|
| 53 |
+
"upsample_type": "conv",
|
| 54 |
+
"use_ema_warmup": true
|
| 55 |
+
}
|
checkpoint-2000/unet_ema/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:85e591ab4548c0e3e0c7fbf1deb7fec7e1e48409be4ab46e15e0038097af13fc
|
| 3 |
+
size 454759548
|
checkpoint-4000/optimizer.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c6b1dbd42dbcfbb76ca7a21fe8cae8c3979be35e551688d51f08bb485743a6bd
|
| 3 |
+
size 909800163
|
checkpoint-4000/random_states_0.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7843dd8cc6b6a32d0207c2fd57a5249aecc6ff6792977b0d5d67ae47f3323b1c
|
| 3 |
+
size 14663
|
checkpoint-4000/scheduler.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f28ee6b328329cfb12d3efaf5269f38a18a88dce2bb3ac4e40db8f327dbb5ae5
|
| 3 |
+
size 563
|
checkpoint-4000/unet/config.json
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "UNet2DModel",
|
| 3 |
+
"_diffusers_version": "0.21.0.dev0",
|
| 4 |
+
"act_fn": "silu",
|
| 5 |
+
"add_attention": true,
|
| 6 |
+
"attention_head_dim": 8,
|
| 7 |
+
"block_out_channels": [
|
| 8 |
+
128,
|
| 9 |
+
128,
|
| 10 |
+
256,
|
| 11 |
+
256,
|
| 12 |
+
512,
|
| 13 |
+
512
|
| 14 |
+
],
|
| 15 |
+
"center_input_sample": false,
|
| 16 |
+
"class_embed_type": null,
|
| 17 |
+
"down_block_types": [
|
| 18 |
+
"DownBlock2D",
|
| 19 |
+
"DownBlock2D",
|
| 20 |
+
"DownBlock2D",
|
| 21 |
+
"DownBlock2D",
|
| 22 |
+
"AttnDownBlock2D",
|
| 23 |
+
"DownBlock2D"
|
| 24 |
+
],
|
| 25 |
+
"downsample_padding": 1,
|
| 26 |
+
"downsample_type": "conv",
|
| 27 |
+
"flip_sin_to_cos": true,
|
| 28 |
+
"freq_shift": 0,
|
| 29 |
+
"in_channels": 5,
|
| 30 |
+
"layers_per_block": 2,
|
| 31 |
+
"mid_block_scale_factor": 1,
|
| 32 |
+
"norm_eps": 1e-05,
|
| 33 |
+
"norm_num_groups": 32,
|
| 34 |
+
"num_class_embeds": null,
|
| 35 |
+
"out_channels": 5,
|
| 36 |
+
"resnet_time_scale_shift": "default",
|
| 37 |
+
"sample_size": 32,
|
| 38 |
+
"time_embedding_type": "positional",
|
| 39 |
+
"up_block_types": [
|
| 40 |
+
"UpBlock2D",
|
| 41 |
+
"AttnUpBlock2D",
|
| 42 |
+
"UpBlock2D",
|
| 43 |
+
"UpBlock2D",
|
| 44 |
+
"UpBlock2D",
|
| 45 |
+
"UpBlock2D"
|
| 46 |
+
],
|
| 47 |
+
"upsample_type": "conv"
|
| 48 |
+
}
|
checkpoint-4000/unet/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06df0e27cc886eb0034765e010e0009beca583ca6ce62f5be078f74243b7b726
|
| 3 |
+
size 454759548
|
checkpoint-4000/unet_ema/config.json
ADDED
|
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "UNet2DModel",
|
| 3 |
+
"_diffusers_version": "0.21.0.dev0",
|
| 4 |
+
"act_fn": "silu",
|
| 5 |
+
"add_attention": true,
|
| 6 |
+
"attention_head_dim": 8,
|
| 7 |
+
"block_out_channels": [
|
| 8 |
+
128,
|
| 9 |
+
128,
|
| 10 |
+
256,
|
| 11 |
+
256,
|
| 12 |
+
512,
|
| 13 |
+
512
|
| 14 |
+
],
|
| 15 |
+
"center_input_sample": false,
|
| 16 |
+
"class_embed_type": null,
|
| 17 |
+
"decay": 0.9999,
|
| 18 |
+
"down_block_types": [
|
| 19 |
+
"DownBlock2D",
|
| 20 |
+
"DownBlock2D",
|
| 21 |
+
"DownBlock2D",
|
| 22 |
+
"DownBlock2D",
|
| 23 |
+
"AttnDownBlock2D",
|
| 24 |
+
"DownBlock2D"
|
| 25 |
+
],
|
| 26 |
+
"downsample_padding": 1,
|
| 27 |
+
"downsample_type": "conv",
|
| 28 |
+
"flip_sin_to_cos": true,
|
| 29 |
+
"freq_shift": 0,
|
| 30 |
+
"in_channels": 5,
|
| 31 |
+
"inv_gamma": 1.0,
|
| 32 |
+
"layers_per_block": 2,
|
| 33 |
+
"mid_block_scale_factor": 1,
|
| 34 |
+
"min_decay": 0.0,
|
| 35 |
+
"norm_eps": 1e-05,
|
| 36 |
+
"norm_num_groups": 32,
|
| 37 |
+
"num_class_embeds": null,
|
| 38 |
+
"optimization_step": 4000,
|
| 39 |
+
"out_channels": 5,
|
| 40 |
+
"power": 0.75,
|
| 41 |
+
"resnet_time_scale_shift": "default",
|
| 42 |
+
"sample_size": 32,
|
| 43 |
+
"time_embedding_type": "positional",
|
| 44 |
+
"up_block_types": [
|
| 45 |
+
"UpBlock2D",
|
| 46 |
+
"AttnUpBlock2D",
|
| 47 |
+
"UpBlock2D",
|
| 48 |
+
"UpBlock2D",
|
| 49 |
+
"UpBlock2D",
|
| 50 |
+
"UpBlock2D"
|
| 51 |
+
],
|
| 52 |
+
"update_after_step": 0,
|
| 53 |
+
"upsample_type": "conv",
|
| 54 |
+
"use_ema_warmup": true
|
| 55 |
+
}
|
checkpoint-4000/unet_ema/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:14447e98eada33506f176220dd3c79b5d6b6c5a91a9c1e0cf8a13c0c5d186267
|
| 3 |
+
size 454759548
|
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 454759548
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:55a3f0597b4bda8aea2734b12f8ec73b62a8c3c1e78ea5c10f179a3d7b53d2cb
|
| 3 |
size 454759548
|