Upload dit4sr-replication from checkpoint-150000
Browse files
transformer/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "SD3Transformer2DModel",
|
| 3 |
+
"_diffusers_version": "0.36.0",
|
| 4 |
+
"_name_or_path": "./experiments/dit4sr/checkpoint-115000",
|
| 5 |
+
"attention_head_dim": 64,
|
| 6 |
+
"caption_projection_dim": 1536,
|
| 7 |
+
"dual_attention_layers": [
|
| 8 |
+
0,
|
| 9 |
+
1,
|
| 10 |
+
2,
|
| 11 |
+
3,
|
| 12 |
+
4,
|
| 13 |
+
5,
|
| 14 |
+
6,
|
| 15 |
+
7,
|
| 16 |
+
8,
|
| 17 |
+
9,
|
| 18 |
+
10,
|
| 19 |
+
11,
|
| 20 |
+
12
|
| 21 |
+
],
|
| 22 |
+
"in_channels": 16,
|
| 23 |
+
"joint_attention_dim": 4096,
|
| 24 |
+
"num_attention_heads": 24,
|
| 25 |
+
"num_layers": 24,
|
| 26 |
+
"out_channels": 16,
|
| 27 |
+
"patch_size": 2,
|
| 28 |
+
"pooled_projection_dim": 2048,
|
| 29 |
+
"pos_embed_max_size": 384,
|
| 30 |
+
"qk_norm": "rms_norm",
|
| 31 |
+
"sample_size": 128
|
| 32 |
+
}
|
transformer/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99a11a73b598603b3af2b34789c38e313ab75b2ef911d0c7ced8794b5b1c6382
|
| 3 |
+
size 6588529984
|