marcelittle commited on
Commit
d759847
·
1 Parent(s): 2997a8a
checkpoint-14160/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd09bba0d81faa458eceaafec40de9896e689c1c36d3b35c3b1912aa194429b0
3
+ size 909763736
checkpoint-14160/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f93da0f8791b69ad2e2fe1a1c1a904ef19766be05ab2ae4e00bad348f54e970d
3
+ size 14344
checkpoint-14160/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96f4f62e2ac271f33d7a4f42480bb34f26a5904c729c95a32db3100f98c39e15
3
+ size 1000
checkpoint-14160/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.24.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 256,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
checkpoint-14160/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d63a2a5088e73abb8b918eca7fbb1be701c18fe2990c8441ab1b6b57bb7e8bf
3
+ size 454741108
checkpoint-14160/unet_ema/config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.24.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
18
+ "decay": 0.9999,
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
+ "inv_gamma": 1.0,
34
+ "layers_per_block": 2,
35
+ "mid_block_scale_factor": 1,
36
+ "min_decay": 0.0,
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "num_train_timesteps": null,
41
+ "optimization_step": 14160,
42
+ "out_channels": 3,
43
+ "power": 0.75,
44
+ "resnet_time_scale_shift": "default",
45
+ "sample_size": 256,
46
+ "time_embedding_type": "positional",
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "AttnUpBlock2D",
50
+ "UpBlock2D",
51
+ "UpBlock2D",
52
+ "UpBlock2D",
53
+ "UpBlock2D"
54
+ ],
55
+ "update_after_step": 0,
56
+ "upsample_type": "conv",
57
+ "use_ema_warmup": true
58
+ }
checkpoint-14160/unet_ema/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2cfa0c5bbda45662f4b7cab6ef57bcae0253aab3996b02388be92c8988715cb7
3
+ size 454741108
epoch-130/model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMPipeline",
3
+ "_diffusers_version": "0.24.0.dev0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDPMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNet2DModel"
11
+ ]
12
+ }
epoch-130/scheduler/scheduler_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.24.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
+ "sample_max_value": 1.0,
13
+ "steps_offset": 0,
14
+ "thresholding": false,
15
+ "timestep_spacing": "leading",
16
+ "trained_betas": null,
17
+ "variance_type": "fixed_small"
18
+ }
epoch-130/unet/config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.24.0.dev0",
4
+ "act_fn": "silu",
5
+ "add_attention": true,
6
+ "attention_head_dim": 8,
7
+ "attn_norm_num_groups": null,
8
+ "block_out_channels": [
9
+ 128,
10
+ 128,
11
+ 256,
12
+ 256,
13
+ 512,
14
+ 512
15
+ ],
16
+ "center_input_sample": false,
17
+ "class_embed_type": null,
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
+ "dropout": 0.0,
29
+ "flip_sin_to_cos": true,
30
+ "freq_shift": 0,
31
+ "in_channels": 3,
32
+ "layers_per_block": 2,
33
+ "mid_block_scale_factor": 1,
34
+ "norm_eps": 1e-05,
35
+ "norm_num_groups": 32,
36
+ "num_class_embeds": null,
37
+ "num_train_timesteps": null,
38
+ "out_channels": 3,
39
+ "resnet_time_scale_shift": "default",
40
+ "sample_size": 256,
41
+ "time_embedding_type": "positional",
42
+ "up_block_types": [
43
+ "UpBlock2D",
44
+ "AttnUpBlock2D",
45
+ "UpBlock2D",
46
+ "UpBlock2D",
47
+ "UpBlock2D",
48
+ "UpBlock2D"
49
+ ],
50
+ "upsample_type": "conv"
51
+ }
epoch-130/unet/diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2eac10b07d5d25030bba4bc3777e9a58000b2dc5a41e1e3f468e64c09b88fb64
3
+ size 454741108
logs/train_unconditional/events.out.tfevents.1702440575.lab.737396.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b9baaa500a83f5254bdf2069825e402b79731b9d10c839749a909b0e343196f9
3
- size 6825448
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e944568586a6c306bbc68a403d3cc21b8632dd25a93167660b203c73b4bd318e
3
+ size 7345731