jonbesga commited on
Commit
6351725
·
verified ·
1 Parent(s): 85d56f3
Files changed (31) hide show
  1. .ipynb_checkpoints/Untitled-checkpoint.ipynb +6 -0
  2. Untitled.ipynb +122 -0
  3. checkpoint-25000/optimizer.bin +3 -0
  4. checkpoint-25000/random_states_0.pkl +3 -0
  5. checkpoint-25000/scaler.pt +3 -0
  6. checkpoint-25000/scheduler.bin +3 -0
  7. checkpoint-25000/unet/config.json +52 -0
  8. checkpoint-25000/unet/diffusion_pytorch_model.safetensors +3 -0
  9. checkpoint-25500/optimizer.bin +3 -0
  10. checkpoint-25500/random_states_0.pkl +3 -0
  11. checkpoint-25500/scaler.pt +3 -0
  12. checkpoint-25500/scheduler.bin +3 -0
  13. checkpoint-25500/unet/config.json +52 -0
  14. checkpoint-25500/unet/diffusion_pytorch_model.safetensors +3 -0
  15. checkpoint-26000/optimizer.bin +3 -0
  16. checkpoint-26000/random_states_0.pkl +3 -0
  17. checkpoint-26000/scaler.pt +3 -0
  18. checkpoint-26000/scheduler.bin +3 -0
  19. checkpoint-26000/unet/config.json +52 -0
  20. checkpoint-26000/unet/diffusion_pytorch_model.safetensors +3 -0
  21. logs/train_unconditional/events.out.tfevents.1724945164.f786777584b2.1108.0 +3 -0
  22. logs/train_unconditional/events.out.tfevents.1724948056.5a8349fd16a5.332.0 +3 -0
  23. logs/train_unconditional/events.out.tfevents.1724948442.5a8349fd16a5.750.0 +3 -0
  24. logs/train_unconditional/events.out.tfevents.1724948466.5a8349fd16a5.828.0 +3 -0
  25. logs/train_unconditional/events.out.tfevents.1724950687.5a8349fd16a5.1146.0 +3 -0
  26. logs/train_unconditional/events.out.tfevents.1724950713.5a8349fd16a5.1227.0 +3 -0
  27. logs/train_unconditional/events.out.tfevents.1724951910.5a8349fd16a5.1677.0 +3 -0
  28. model_index.json +12 -0
  29. scheduler/scheduler_config.json +19 -0
  30. unet/config.json +52 -0
  31. unet/diffusion_pytorch_model.safetensors +3 -0
.ipynb_checkpoints/Untitled-checkpoint.ipynb ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "cells": [],
3
+ "metadata": {},
4
+ "nbformat": 4,
5
+ "nbformat_minor": 5
6
+ }
Untitled.ipynb ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cells": [
3
+ {
4
+ "cell_type": "code",
5
+ "execution_count": 3,
6
+ "id": "f7c2711d-581e-433d-b516-375f663b512b",
7
+ "metadata": {},
8
+ "outputs": [
9
+ {
10
+ "name": "stdout",
11
+ "output_type": "stream",
12
+ "text": [
13
+ "Requirement already satisfied: tensorboard in /usr/local/lib/python3.11/dist-packages (2.17.1)\n",
14
+ "Requirement already satisfied: absl-py>=0.4 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (2.1.0)\n",
15
+ "Requirement already satisfied: grpcio>=1.48.2 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (1.66.1)\n",
16
+ "Requirement already satisfied: markdown>=2.6.8 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (3.7)\n",
17
+ "Requirement already satisfied: numpy>=1.12.0 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (1.26.4)\n",
18
+ "Requirement already satisfied: packaging in /usr/local/lib/python3.11/dist-packages (from tensorboard) (24.1)\n",
19
+ "Requirement already satisfied: protobuf!=4.24.0,>=3.19.6 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (5.28.0)\n",
20
+ "Requirement already satisfied: setuptools>=41.0.0 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (73.0.1)\n",
21
+ "Requirement already satisfied: six>1.9 in /usr/lib/python3/dist-packages (from tensorboard) (1.16.0)\n",
22
+ "Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (0.7.2)\n",
23
+ "Requirement already satisfied: werkzeug>=1.0.1 in /usr/local/lib/python3.11/dist-packages (from tensorboard) (3.0.4)\n",
24
+ "Requirement already satisfied: MarkupSafe>=2.1.1 in /usr/local/lib/python3.11/dist-packages (from werkzeug>=1.0.1->tensorboard) (2.1.5)\n",
25
+ "\u001b[33mWARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning.\u001b[0m\u001b[33m\n",
26
+ "\u001b[0m"
27
+ ]
28
+ }
29
+ ],
30
+ "source": [
31
+ "!pip install tensorboard"
32
+ ]
33
+ },
34
+ {
35
+ "cell_type": "code",
36
+ "execution_count": 6,
37
+ "id": "3405637e-767e-47ac-b2ed-2fad64da0c32",
38
+ "metadata": {},
39
+ "outputs": [
40
+ {
41
+ "name": "stdout",
42
+ "output_type": "stream",
43
+ "text": [
44
+ "The tensorboard extension is already loaded. To reload it, use:\n",
45
+ " %reload_ext tensorboard\n"
46
+ ]
47
+ }
48
+ ],
49
+ "source": [
50
+ "%load_ext tensorboard"
51
+ ]
52
+ },
53
+ {
54
+ "cell_type": "code",
55
+ "execution_count": 11,
56
+ "id": "8589581d-09e6-4a30-8a3e-606350df6f30",
57
+ "metadata": {},
58
+ "outputs": [],
59
+ "source": [
60
+ "!kill 1448"
61
+ ]
62
+ },
63
+ {
64
+ "cell_type": "code",
65
+ "execution_count": 12,
66
+ "id": "5404abf4-c654-48d6-b1cb-bb60b35620ad",
67
+ "metadata": {},
68
+ "outputs": [
69
+ {
70
+ "data": {
71
+ "text/html": [
72
+ "\n",
73
+ " <iframe id=\"tensorboard-frame-6028da6349f13372\" width=\"100%\" height=\"800\" frameborder=\"0\">\n",
74
+ " </iframe>\n",
75
+ " <script>\n",
76
+ " (function() {\n",
77
+ " const frame = document.getElementById(\"tensorboard-frame-6028da6349f13372\");\n",
78
+ " const url = new URL(\"/\", window.location);\n",
79
+ " const port = 6007;\n",
80
+ " if (port) {\n",
81
+ " url.port = port;\n",
82
+ " }\n",
83
+ " frame.src = url;\n",
84
+ " })();\n",
85
+ " </script>\n",
86
+ " "
87
+ ],
88
+ "text/plain": [
89
+ "<IPython.core.display.HTML object>"
90
+ ]
91
+ },
92
+ "metadata": {},
93
+ "output_type": "display_data"
94
+ }
95
+ ],
96
+ "source": [
97
+ "%tensorboard --logdir=\"logs/train_unconditional\""
98
+ ]
99
+ }
100
+ ],
101
+ "metadata": {
102
+ "kernelspec": {
103
+ "display_name": "Python 3 (ipykernel)",
104
+ "language": "python",
105
+ "name": "python3"
106
+ },
107
+ "language_info": {
108
+ "codemirror_mode": {
109
+ "name": "ipython",
110
+ "version": 3
111
+ },
112
+ "file_extension": ".py",
113
+ "mimetype": "text/x-python",
114
+ "name": "python",
115
+ "nbconvert_exporter": "python",
116
+ "pygments_lexer": "ipython3",
117
+ "version": "3.11.9"
118
+ }
119
+ },
120
+ "nbformat": 4,
121
+ "nbformat_minor": 5
122
+ }
checkpoint-25000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c483847f331a7e2f01eb64d6191f8b60d4fbadba809e679d93be86f10a9bf92
3
+ size 909763736
checkpoint-25000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:964c4bd89dca5e798414a1c5998aa20bf186a000703c11a551a73bbb39abc455
3
+ size 14344
checkpoint-25000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3902407c0bdba642188979b6f3ec61116e8beb34c3409ade0052efd965392245
3
+ size 988
checkpoint-25000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a15d7fc66f1b75f6fad84e1585257d21304771763410529e0386cc96c099db4
3
+ size 1000
checkpoint-25000/unet/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.31.0.dev0",
4
+ "_name_or_path": "ddpm-ema-flowers-64/checkpoint-20500",
5
+ "act_fn": "silu",
6
+ "add_attention": true,
7
+ "attention_head_dim": 8,
8
+ "attn_norm_num_groups": null,
9
+ "block_out_channels": [
10
+ 128,
11
+ 128,
12
+ 256,
13
+ 256,
14
+ 512,
15
+ 512
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": null,
38
+ "num_train_timesteps": null,
39
+ "out_channels": 3,
40
+ "resnet_time_scale_shift": "default",
41
+ "sample_size": 64,
42
+ "time_embedding_type": "positional",
43
+ "up_block_types": [
44
+ "UpBlock2D",
45
+ "AttnUpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D"
50
+ ],
51
+ "upsample_type": "conv"
52
+ }
checkpoint-25000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:070ca441d1b75265e316b7d55aae1868319f4380f86c1774e024ed49cc2e23bb
3
+ size 454741108
checkpoint-25500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e1ea30b1b4571763154fc7d86bed67ba12a8cc75f8d127d3a349f21d419fe821
3
+ size 909763736
checkpoint-25500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0805005f2e2a1a0d32ec954b5d9bde446e2f2b919e0f88a64936888ed8d61c35
3
+ size 14344
checkpoint-25500/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2d690134e13348a3157a106686199237e3f618e08cd8ff24deb4f011dd86132
3
+ size 988
checkpoint-25500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b0fa01b8981b124d8675447516896509db140df8b566d2fbe73520f4f67651d
3
+ size 1000
checkpoint-25500/unet/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.31.0.dev0",
4
+ "_name_or_path": "ddpm-ema-flowers-64/checkpoint-20500",
5
+ "act_fn": "silu",
6
+ "add_attention": true,
7
+ "attention_head_dim": 8,
8
+ "attn_norm_num_groups": null,
9
+ "block_out_channels": [
10
+ 128,
11
+ 128,
12
+ 256,
13
+ 256,
14
+ 512,
15
+ 512
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": null,
38
+ "num_train_timesteps": null,
39
+ "out_channels": 3,
40
+ "resnet_time_scale_shift": "default",
41
+ "sample_size": 64,
42
+ "time_embedding_type": "positional",
43
+ "up_block_types": [
44
+ "UpBlock2D",
45
+ "AttnUpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D"
50
+ ],
51
+ "upsample_type": "conv"
52
+ }
checkpoint-25500/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:258a1c40a3e26a0e5e7702f2b015e60812c750d9a9f429da81a52a2703a246c3
3
+ size 454741108
checkpoint-26000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d34f9155665b0f49600ea9902868fa4128acdd703d57466236793deb2290873
3
+ size 909763736
checkpoint-26000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5230f3421d96d0dbc2c3106d593079594d3943f79cc327548648deb7e4b2cab4
3
+ size 14344
checkpoint-26000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9d10e361c83d25f72cca0a8e57e25b46c003102ecb4b4df48049446c3a97288
3
+ size 988
checkpoint-26000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4aa902efa8e0fd93a4a483339b54a4d7ab9743474d841e3ba71991893981c3d
3
+ size 1000
checkpoint-26000/unet/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.31.0.dev0",
4
+ "_name_or_path": "ddpm-ema-flowers-64/checkpoint-20500",
5
+ "act_fn": "silu",
6
+ "add_attention": true,
7
+ "attention_head_dim": 8,
8
+ "attn_norm_num_groups": null,
9
+ "block_out_channels": [
10
+ 128,
11
+ 128,
12
+ 256,
13
+ 256,
14
+ 512,
15
+ 512
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": null,
38
+ "num_train_timesteps": null,
39
+ "out_channels": 3,
40
+ "resnet_time_scale_shift": "default",
41
+ "sample_size": 64,
42
+ "time_embedding_type": "positional",
43
+ "up_block_types": [
44
+ "UpBlock2D",
45
+ "AttnUpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D"
50
+ ],
51
+ "upsample_type": "conv"
52
+ }
checkpoint-26000/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5488902dd2f1d0dcb04b009f7747ebbd56c07268cba58baa8cea4af38d2e7bf
3
+ size 454741108
logs/train_unconditional/events.out.tfevents.1724945164.f786777584b2.1108.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d2a18889df89153df8d07d02223286729240557443ad8b590ad377d6d4fb76f
3
+ size 2266528
logs/train_unconditional/events.out.tfevents.1724948056.5a8349fd16a5.332.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d100c0b3c6051949bcf63639415d68c30ff77c73f9dbf784664607a01af05f6
3
+ size 126961
logs/train_unconditional/events.out.tfevents.1724948442.5a8349fd16a5.750.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a905b623bb3250b3e21d5b4c39e339f4925316aefc38039d995b99348672f495
3
+ size 88
logs/train_unconditional/events.out.tfevents.1724948466.5a8349fd16a5.828.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b92a8962b7dda2f4f87e9b26717a448383c56b23260443108b05a48c1ef7e0c7
3
+ size 1093201
logs/train_unconditional/events.out.tfevents.1724950687.5a8349fd16a5.1146.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44d6cb197753d2f7cac1408808368c4dfbaf8f07a6bace4cf497cf4444d5dd01
3
+ size 88
logs/train_unconditional/events.out.tfevents.1724950713.5a8349fd16a5.1227.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a453e68dce22c347f57d7c5d0fb56b7b51f3e30382f7ac11c2ced0fbae4bc174
3
+ size 853595
logs/train_unconditional/events.out.tfevents.1724951910.5a8349fd16a5.1677.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e196646f446d187c009e81e58322313815d6967c9b83e217329f40705a02700
3
+ size 153386
model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMPipeline",
3
+ "_diffusers_version": "0.31.0.dev0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDPMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNet2DModel"
11
+ ]
12
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.31.0.dev0",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "linear",
6
+ "beta_start": 0.0001,
7
+ "clip_sample": true,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 1000,
11
+ "prediction_type": "epsilon",
12
+ "rescale_betas_zero_snr": false,
13
+ "sample_max_value": 1.0,
14
+ "steps_offset": 0,
15
+ "thresholding": false,
16
+ "timestep_spacing": "leading",
17
+ "trained_betas": null,
18
+ "variance_type": "fixed_small"
19
+ }
unet/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.31.0.dev0",
4
+ "_name_or_path": "ddpm-ema-flowers-64/checkpoint-26000",
5
+ "act_fn": "silu",
6
+ "add_attention": true,
7
+ "attention_head_dim": 8,
8
+ "attn_norm_num_groups": null,
9
+ "block_out_channels": [
10
+ 128,
11
+ 128,
12
+ 256,
13
+ 256,
14
+ 512,
15
+ 512
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "down_block_types": [
20
+ "DownBlock2D",
21
+ "DownBlock2D",
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "AttnDownBlock2D",
25
+ "DownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "downsample_type": "conv",
29
+ "dropout": 0.0,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 3,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "norm_eps": 1e-05,
36
+ "norm_num_groups": 32,
37
+ "num_class_embeds": null,
38
+ "num_train_timesteps": null,
39
+ "out_channels": 3,
40
+ "resnet_time_scale_shift": "default",
41
+ "sample_size": 64,
42
+ "time_embedding_type": "positional",
43
+ "up_block_types": [
44
+ "UpBlock2D",
45
+ "AttnUpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D",
49
+ "UpBlock2D"
50
+ ],
51
+ "upsample_type": "conv"
52
+ }
unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:909afcff5a81f47f1be0868f765a18da52bcafa184e5749dfad1e5a1b5444490
3
+ size 454741108