adammoss commited on
Commit
c3f3bc9
1 Parent(s): 70a1f9a
checkpoint-40000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a8e224025c7cde961bd59dbc395158c2c97b42a0dbd0fd51034de2ea6d96e30
3
+ size 227868455
checkpoint-40000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08119afac46acdd99179be572c3a749c60dfed79a6b3f3802cdf68f7fb019b16
3
+ size 14663
checkpoint-40000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46f0dee12ccd81961112619ce8c3f5e990f1d7b45fa215a762de7640c207e7ff
3
+ size 563
checkpoint-40000/unet/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 128,
11
+ 192,
12
+ 256
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "down_block_types": [
22
+ "DownBlock2D",
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "CrossAttnDownBlock2D"
26
+ ],
27
+ "downsample_padding": 1,
28
+ "dual_cross_attention": false,
29
+ "encoder_hid_dim": 7,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 2,
33
+ "layers_per_block": 2,
34
+ "mid_block_only_cross_attention": null,
35
+ "mid_block_scale_factor": 1,
36
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
+ "norm_num_groups": 32,
39
+ "num_class_embeds": null,
40
+ "only_cross_attention": false,
41
+ "out_channels": 1,
42
+ "projection_class_embeddings_input_dim": null,
43
+ "resnet_out_scale_factor": 1.0,
44
+ "resnet_skip_time_act": false,
45
+ "resnet_time_scale_shift": "default",
46
+ "sample_size": 256,
47
+ "time_cond_proj_dim": null,
48
+ "time_embedding_act_fn": null,
49
+ "time_embedding_dim": null,
50
+ "time_embedding_type": "positional",
51
+ "timestep_post_act": null,
52
+ "up_block_types": [
53
+ "CrossAttnUpBlock2D",
54
+ "UpBlock2D",
55
+ "UpBlock2D",
56
+ "UpBlock2D"
57
+ ],
58
+ "upcast_attention": false,
59
+ "use_linear_projection": false
60
+ }
checkpoint-40000/unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f26e39d4a24c88bba19ae6607786ef7168092f632d37dc56531b72e280f4cf39
3
+ size 113930213
checkpoint-40000/unet_ema/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.16.1",
4
+ "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
+ "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 64,
10
+ 128,
11
+ 192,
12
+ 256
13
+ ],
14
+ "center_input_sample": false,
15
+ "class_embed_type": null,
16
+ "class_embeddings_concat": false,
17
+ "conv_in_kernel": 3,
18
+ "conv_out_kernel": 3,
19
+ "cross_attention_dim": 256,
20
+ "cross_attention_norm": null,
21
+ "decay": 0.9999,
22
+ "down_block_types": [
23
+ "DownBlock2D",
24
+ "DownBlock2D",
25
+ "DownBlock2D",
26
+ "CrossAttnDownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "encoder_hid_dim": 7,
31
+ "flip_sin_to_cos": true,
32
+ "freq_shift": 0,
33
+ "in_channels": 2,
34
+ "inv_gamma": 1.0,
35
+ "layers_per_block": 2,
36
+ "mid_block_only_cross_attention": null,
37
+ "mid_block_scale_factor": 1,
38
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
39
+ "min_decay": 0.0,
40
+ "norm_eps": 1e-05,
41
+ "norm_num_groups": 32,
42
+ "num_class_embeds": null,
43
+ "only_cross_attention": false,
44
+ "optimization_step": 40000,
45
+ "out_channels": 1,
46
+ "power": 0.75,
47
+ "projection_class_embeddings_input_dim": null,
48
+ "resnet_out_scale_factor": 1.0,
49
+ "resnet_skip_time_act": false,
50
+ "resnet_time_scale_shift": "default",
51
+ "sample_size": 256,
52
+ "time_cond_proj_dim": null,
53
+ "time_embedding_act_fn": null,
54
+ "time_embedding_dim": null,
55
+ "time_embedding_type": "positional",
56
+ "timestep_post_act": null,
57
+ "up_block_types": [
58
+ "CrossAttnUpBlock2D",
59
+ "UpBlock2D",
60
+ "UpBlock2D",
61
+ "UpBlock2D"
62
+ ],
63
+ "upcast_attention": false,
64
+ "update_after_step": 0,
65
+ "use_ema_warmup": true,
66
+ "use_linear_projection": false
67
+ }
checkpoint-40000/unet_ema/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c778bd8f632346cfb6aef84ed2d4b587687dd506b4f76e34a466a58e289a4611
3
+ size 113924197
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6d619cc9947d73325c7ee2c634ce351d19e372bb43a4ee732541c252a4145ef1
3
  size 113930213
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd950435238ea4c3a4934ff0a8497d0439e608de36b0bb38d924f565d4518be6
3
  size 113930213