rcannizzaro commited on
Commit
1f81c08
1 Parent(s): 8718756

End of training

Browse files
README.md CHANGED
@@ -7,6 +7,11 @@ tags:
7
  - text-to-image
8
  - diffusers
9
  - diffusers-training
 
 
 
 
 
10
  base_model: runwayml/stable-diffusion-v1-5
11
  inference: true
12
  ---
 
7
  - text-to-image
8
  - diffusers
9
  - diffusers-training
10
+ - stable-diffusion
11
+ - stable-diffusion-diffusers
12
+ - text-to-image
13
+ - diffusers
14
+ - diffusers-training
15
  base_model: runwayml/stable-diffusion-v1-5
16
  inference: true
17
  ---
checkpoint-100/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b4fcd2fb7cdebd5651f0e64cb65127893657d77a800c686233e2b950d6b5ff0
3
  size 6876750164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0946238db3681c0d5a1046c3646b69ccc9bf8dd67dc5b9a4219133e08c14bc66
3
  size 6876750164
checkpoint-100/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a8414fed04d1537441bca7dd68d9d208109d3470204fd3a8fa7a5c2a1eed3728
3
  size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5c608df982f6377dbf9600983ce97818ddef94a366550c656edf92c01528cba
3
  size 14344
checkpoint-100/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cebb8e346451d2bc0ed57568f79ad2bcb6fc0b11caacb4e7ad92ebbeaaa27e2b
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02d2c968708240244814030a78365220059f16bfdc47042bdb4a97a51702f151
3
  size 3438167536
checkpoint-100/unet_ema/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f088d601d438d58adec98c94c72353faab77f60cf5b6f40b68283a37a906104b
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:638cbc3075b9efd7c35497d9733fd9643b5844f68a0efa1f74bbe4915697607c
3
  size 3438167536
checkpoint-3250/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05c7d16bda3ab3304f163f3f5eda7576da368ffd62dfad4aaa267b6381e23674
3
+ size 6876750164
checkpoint-3250/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba06229f087ecda0218291d798de8ee75d5a6dd115a6185e21ed17a906929e5b
3
+ size 14344
checkpoint-3250/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1e263f7f0ea32081b3c38746e64041914f8ca03a2e095097eb8a9fa12fbca36
3
+ size 988
checkpoint-3250/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f3d6989546f7a5c32070e7d40cb0cc5d1f57011d3c25a8500d36b7ab4224e4a1
3
+ size 1000
checkpoint-3250/unet/config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 4,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "reverse_transformer_layers_per_block": null,
53
+ "sample_size": 64,
54
+ "time_cond_proj_dim": null,
55
+ "time_embedding_act_fn": null,
56
+ "time_embedding_dim": null,
57
+ "time_embedding_type": "positional",
58
+ "timestep_post_act": null,
59
+ "transformer_layers_per_block": 1,
60
+ "up_block_types": [
61
+ "UpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D",
64
+ "CrossAttnUpBlock2D"
65
+ ],
66
+ "upcast_attention": false,
67
+ "use_linear_projection": false
68
+ }
checkpoint-3250/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e101f0b935ae5a261022b0d6d7b673df61d303c263447ddfa879872df519f1a8
3
+ size 3438167536
checkpoint-3250/unet_ema/config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "decay": 0.9999,
25
+ "down_block_types": [
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "CrossAttnDownBlock2D",
29
+ "DownBlock2D"
30
+ ],
31
+ "downsample_padding": 1,
32
+ "dropout": 0.0,
33
+ "dual_cross_attention": false,
34
+ "encoder_hid_dim": null,
35
+ "encoder_hid_dim_type": null,
36
+ "flip_sin_to_cos": true,
37
+ "freq_shift": 0,
38
+ "in_channels": 4,
39
+ "inv_gamma": 1.0,
40
+ "layers_per_block": 2,
41
+ "mid_block_only_cross_attention": null,
42
+ "mid_block_scale_factor": 1,
43
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
44
+ "min_decay": 0.0,
45
+ "norm_eps": 1e-05,
46
+ "norm_num_groups": 32,
47
+ "num_attention_heads": null,
48
+ "num_class_embeds": null,
49
+ "only_cross_attention": false,
50
+ "optimization_step": 3250,
51
+ "out_channels": 4,
52
+ "power": 0.6666666666666666,
53
+ "projection_class_embeddings_input_dim": null,
54
+ "resnet_out_scale_factor": 1.0,
55
+ "resnet_skip_time_act": false,
56
+ "resnet_time_scale_shift": "default",
57
+ "reverse_transformer_layers_per_block": null,
58
+ "sample_size": 64,
59
+ "time_cond_proj_dim": null,
60
+ "time_embedding_act_fn": null,
61
+ "time_embedding_dim": null,
62
+ "time_embedding_type": "positional",
63
+ "timestep_post_act": null,
64
+ "transformer_layers_per_block": 1,
65
+ "up_block_types": [
66
+ "UpBlock2D",
67
+ "CrossAttnUpBlock2D",
68
+ "CrossAttnUpBlock2D",
69
+ "CrossAttnUpBlock2D"
70
+ ],
71
+ "upcast_attention": false,
72
+ "update_after_step": 0,
73
+ "use_ema_warmup": false,
74
+ "use_linear_projection": false
75
+ }
checkpoint-3250/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45b12fba9e1e4cb8107c9eb3f22e4c719c3cf8e079a394f423420738b5124eab
3
+ size 3438167536
checkpoint-3300/unet/config.json ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "down_block_types": [
25
+ "CrossAttnDownBlock2D",
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "DownBlock2D"
29
+ ],
30
+ "downsample_padding": 1,
31
+ "dropout": 0.0,
32
+ "dual_cross_attention": false,
33
+ "encoder_hid_dim": null,
34
+ "encoder_hid_dim_type": null,
35
+ "flip_sin_to_cos": true,
36
+ "freq_shift": 0,
37
+ "in_channels": 4,
38
+ "layers_per_block": 2,
39
+ "mid_block_only_cross_attention": null,
40
+ "mid_block_scale_factor": 1,
41
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
42
+ "norm_eps": 1e-05,
43
+ "norm_num_groups": 32,
44
+ "num_attention_heads": null,
45
+ "num_class_embeds": null,
46
+ "only_cross_attention": false,
47
+ "out_channels": 4,
48
+ "projection_class_embeddings_input_dim": null,
49
+ "resnet_out_scale_factor": 1.0,
50
+ "resnet_skip_time_act": false,
51
+ "resnet_time_scale_shift": "default",
52
+ "reverse_transformer_layers_per_block": null,
53
+ "sample_size": 64,
54
+ "time_cond_proj_dim": null,
55
+ "time_embedding_act_fn": null,
56
+ "time_embedding_dim": null,
57
+ "time_embedding_type": "positional",
58
+ "timestep_post_act": null,
59
+ "transformer_layers_per_block": 1,
60
+ "up_block_types": [
61
+ "UpBlock2D",
62
+ "CrossAttnUpBlock2D",
63
+ "CrossAttnUpBlock2D",
64
+ "CrossAttnUpBlock2D"
65
+ ],
66
+ "upcast_attention": false,
67
+ "use_linear_projection": false
68
+ }
checkpoint-3300/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:277e1a25806c734b9ea519775c23dfb387f29d1de955aaf485af1ea00433fd00
3
+ size 3267026944
checkpoint-3300/unet_ema/config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "runwayml/stable-diffusion-v1-5",
5
+ "act_fn": "silu",
6
+ "addition_embed_type": null,
7
+ "addition_embed_type_num_heads": 64,
8
+ "addition_time_embed_dim": null,
9
+ "attention_head_dim": 8,
10
+ "attention_type": "default",
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "class_embeddings_concat": false,
20
+ "conv_in_kernel": 3,
21
+ "conv_out_kernel": 3,
22
+ "cross_attention_dim": 768,
23
+ "cross_attention_norm": null,
24
+ "decay": 0.9999,
25
+ "down_block_types": [
26
+ "CrossAttnDownBlock2D",
27
+ "CrossAttnDownBlock2D",
28
+ "CrossAttnDownBlock2D",
29
+ "DownBlock2D"
30
+ ],
31
+ "downsample_padding": 1,
32
+ "dropout": 0.0,
33
+ "dual_cross_attention": false,
34
+ "encoder_hid_dim": null,
35
+ "encoder_hid_dim_type": null,
36
+ "flip_sin_to_cos": true,
37
+ "freq_shift": 0,
38
+ "in_channels": 4,
39
+ "inv_gamma": 1.0,
40
+ "layers_per_block": 2,
41
+ "mid_block_only_cross_attention": null,
42
+ "mid_block_scale_factor": 1,
43
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
44
+ "min_decay": 0.0,
45
+ "norm_eps": 1e-05,
46
+ "norm_num_groups": 32,
47
+ "num_attention_heads": null,
48
+ "num_class_embeds": null,
49
+ "only_cross_attention": false,
50
+ "optimization_step": 3300,
51
+ "out_channels": 4,
52
+ "power": 0.6666666666666666,
53
+ "projection_class_embeddings_input_dim": null,
54
+ "resnet_out_scale_factor": 1.0,
55
+ "resnet_skip_time_act": false,
56
+ "resnet_time_scale_shift": "default",
57
+ "reverse_transformer_layers_per_block": null,
58
+ "sample_size": 64,
59
+ "time_cond_proj_dim": null,
60
+ "time_embedding_act_fn": null,
61
+ "time_embedding_dim": null,
62
+ "time_embedding_type": "positional",
63
+ "timestep_post_act": null,
64
+ "transformer_layers_per_block": 1,
65
+ "up_block_types": [
66
+ "UpBlock2D",
67
+ "CrossAttnUpBlock2D",
68
+ "CrossAttnUpBlock2D",
69
+ "CrossAttnUpBlock2D"
70
+ ],
71
+ "upcast_attention": false,
72
+ "update_after_step": 0,
73
+ "use_ema_warmup": false,
74
+ "use_linear_projection": false
75
+ }
checkpoint-3300/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88e14aa19648efbc3d1b88a898101360861f5357834f956441182a005f2a2822
3
+ size 3438167536
checkpoint-50/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5cdc718a49e5fa698196edddb901f6c9ad9e06c38f366627fc2555169e522c48
3
  size 6876750164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c541103643b3e69fa294d794b81851b1dc462d8fe2d28244c173308127b1d10
3
  size 6876750164
checkpoint-50/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5df4a55bc15ff21055fefb7a540be14b699d9c6fe3e89063dcc018df4898f625
3
  size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d9b763e6c5c08ffb00a1e405ccb3453584ac891295b178e12d5c5f2ea44a46b
3
  size 14344
checkpoint-50/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ca763ac31735f299e755f3c46af03bd00e9ae5795f4f7de2d7efe822b8a8d0ea
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30b7baf09c9e94f31be6dcd97ea617335704a7461e945a884c626c57474aa8c7
3
  size 3438167536
checkpoint-50/unet_ema/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7788af3ab2920f6d2b1aa485da8aa19e4d365ad9330bf2c3e96341e00f1c7150
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e40d7a12697110f9538915c05702b1617ee21877ebcf0d3fb8985490c8c3da0b
3
  size 3438167536
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f088d601d438d58adec98c94c72353faab77f60cf5b6f40b68283a37a906104b
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:638cbc3075b9efd7c35497d9733fd9643b5844f68a0efa1f74bbe4915697607c
3
  size 3438167536
vae/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "DSpritesSingleChannelVAE",
3
  "_diffusers_version": "0.30.0.dev0",
4
- "_name_or_path": "CompVis/stable-diffusion-v1-4",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
@@ -16,14 +16,14 @@
16
  "DownEncoderBlock2D"
17
  ],
18
  "force_upcast": true,
19
- "in_channels": 3,
20
  "latent_channels": 4,
21
  "latents_mean": null,
22
  "latents_std": null,
23
  "layers_per_block": 2,
24
  "norm_num_groups": 32,
25
- "out_channels": 3,
26
- "sample_size": 512,
27
  "scaling_factor": 0.18215,
28
  "shift_factor": null,
29
  "up_block_types": [
 
1
  {
2
  "_class_name": "DSpritesSingleChannelVAE",
3
  "_diffusers_version": "0.30.0.dev0",
4
+ "_name_or_path": "rcannizzaro/vae-dsprites",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
16
  "DownEncoderBlock2D"
17
  ],
18
  "force_upcast": true,
19
+ "in_channels": 1,
20
  "latent_channels": 4,
21
  "latents_mean": null,
22
  "latents_std": null,
23
  "layers_per_block": 2,
24
  "norm_num_groups": 32,
25
+ "out_channels": 1,
26
+ "sample_size": 64,
27
  "scaling_factor": 0.18215,
28
  "shift_factor": null,
29
  "up_block_types": [
val_imgs_grid.png CHANGED