tolgacangoz commited on
Commit
a34762a
·
verified ·
1 Parent(s): 860aebc

Update `nesting_level_1` checkpoints

Browse files
unet/nesting_level_1/config.json CHANGED
@@ -5,7 +5,6 @@
5
  "addition_embed_type": "matryoshka",
6
  "addition_embed_type_num_heads": 64,
7
  "addition_time_embed_dim": null,
8
- "args": "<class 'inspect._empty'>",
9
  "attention_head_dim": 8,
10
  "attention_pre_only": false,
11
  "attention_type": "default",
@@ -23,8 +22,8 @@
23
  "cross_attention_norm": null,
24
  "down_block_types": [
25
  "DownBlock2D",
26
- "CrossAttnDownBlock2D",
27
- "CrossAttnDownBlock2D"
28
  ],
29
  "downsample_padding": 1,
30
  "dropout": 0.0,
@@ -52,8 +51,14 @@
52
  "encoder_hid_dim_type": "text_proj",
53
  "flip_sin_to_cos": false,
54
  "in_channels": 3,
 
 
 
 
 
55
  "masked_cross_attention": false,
56
  "micro_conditioning_scale": 64,
 
57
  "nesting": true,
58
  "out_channels": 3,
59
  "resnet_time_scale_shift": "scale_shift",
@@ -66,11 +71,10 @@
66
  "CrossAttnUpBlock2D",
67
  "CrossAttnUpBlock2D",
68
  "UpBlock2D"
69
- ]
 
70
  },
71
- "is_temporal": [
72
- false
73
- ],
74
  "layers_per_block": [
75
  2,
76
  2,
@@ -81,9 +85,7 @@
81
  "mid_block_only_cross_attention": null,
82
  "mid_block_scale_factor": 1,
83
  "mid_block_type": null,
84
- "nest_ratio": [
85
- 4
86
- ],
87
  "nesting": false,
88
  "norm_eps": 1e-05,
89
  "norm_num_groups": 32,
@@ -100,6 +102,7 @@
100
  "sample_size": null,
101
  "skip_cond_emb": false,
102
  "skip_inner_unet_input": false,
 
103
  "temporal_mode": false,
104
  "temporal_spatial_ds": false,
105
  "time_cond_proj_dim": null,
@@ -113,10 +116,11 @@
113
  0
114
  ],
115
  "up_block_types": [
116
- "CrossAttnUpBlock2D",
117
- "CrossAttnUpBlock2D",
118
  "UpBlock2D"
119
  ],
120
  "upcast_attention": false,
 
121
  "use_linear_projection": false
122
  }
 
5
  "addition_embed_type": "matryoshka",
6
  "addition_embed_type_num_heads": 64,
7
  "addition_time_embed_dim": null,
 
8
  "attention_head_dim": 8,
9
  "attention_pre_only": false,
10
  "attention_type": "default",
 
22
  "cross_attention_norm": null,
23
  "down_block_types": [
24
  "DownBlock2D",
25
+ "DownBlock2D",
26
+ "DownBlock2D"
27
  ],
28
  "downsample_padding": 1,
29
  "dropout": 0.0,
 
51
  "encoder_hid_dim_type": "text_proj",
52
  "flip_sin_to_cos": false,
53
  "in_channels": 3,
54
+ "layers_per_block": [
55
+ 2,
56
+ 2,
57
+ 2
58
+ ],
59
  "masked_cross_attention": false,
60
  "micro_conditioning_scale": 64,
61
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
62
  "nesting": true,
63
  "out_channels": 3,
64
  "resnet_time_scale_shift": "scale_shift",
 
71
  "CrossAttnUpBlock2D",
72
  "CrossAttnUpBlock2D",
73
  "UpBlock2D"
74
+ ],
75
+ "use_attention_ffn": true
76
  },
77
+ "is_temporal": null,
 
 
78
  "layers_per_block": [
79
  2,
80
  2,
 
85
  "mid_block_only_cross_attention": null,
86
  "mid_block_scale_factor": 1,
87
  "mid_block_type": null,
88
+ "nest_ratio": null,
 
 
89
  "nesting": false,
90
  "norm_eps": 1e-05,
91
  "norm_num_groups": 32,
 
102
  "sample_size": null,
103
  "skip_cond_emb": false,
104
  "skip_inner_unet_input": false,
105
+ "skip_normalization": true,
106
  "temporal_mode": false,
107
  "temporal_spatial_ds": false,
108
  "time_cond_proj_dim": null,
 
116
  0
117
  ],
118
  "up_block_types": [
119
+ "UpBlock2D",
120
+ "UpBlock2D",
121
  "UpBlock2D"
122
  ],
123
  "upcast_attention": false,
124
+ "use_attention_ffn": false,
125
  "use_linear_projection": false
126
  }
unet/nesting_level_1/diffusion_pytorch_model.fp16.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:86f4c4731b9c0182a9278ceaf10c28b72567f26219a7fbeb10babe8c146d42ea
3
  size 953346068
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d812aa4f1301efcd67892a1c4719f1986793295c0d586781a7578edf996e399
3
  size 953346068
unet/nesting_level_1/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a214799f4584a3bddf326ca574e0b6dea7de83a84a2028c40b2f6b458c8f61b1
3
  size 1906574992
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ceaad1f46833005d2e1a92cfc9fa7ba379d7f535e1c65249a08723c3b96cefde
3
  size 1906574992