nitrofusion2-dit-refiners / tensor_mapping.json
gatilin's picture
Add files using upload-large-folder tool
8a653bd verified
Raw History Blame Contribute Delete
14.3 kB
{
"context_refiner.0.attention.norm_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
22645888,
22646144
],
"dtype": "BF16",
"shape": [
128
]
},
"context_refiner.0.attention.norm_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
22646144,
22646400
],
"dtype": "BF16",
"shape": [
128
]
},
"context_refiner.0.attention.to_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
22646400,
52137600
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.0.attention.to_out.0.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
52137600,
81628800
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.0.attention.to_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
81628800,
111120000
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.0.attention.to_v.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
111120000,
140611200
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.0.attention_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
140611200,
140618880
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.0.attention_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
140618880,
140626560
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.0.feed_forward.w1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
140626560,
219269760
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"context_refiner.0.feed_forward.w2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
219269760,
297912960
],
"dtype": "BF16",
"shape": [
3840,
10240
]
},
"context_refiner.0.feed_forward.w3.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
297912960,
376556160
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"context_refiner.0.ffn_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
376556160,
376563840
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.0.ffn_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
376563840,
376571520
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.1.attention.norm_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
376571520,
376571776
],
"dtype": "BF16",
"shape": [
128
]
},
"context_refiner.1.attention.norm_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
376571776,
376572032
],
"dtype": "BF16",
"shape": [
128
]
},
"context_refiner.1.attention.to_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
376572032,
406063232
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.1.attention.to_out.0.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
406063232,
435554432
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.1.attention.to_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
435554432,
465045632
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.1.attention.to_v.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
465045632,
494536832
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"context_refiner.1.attention_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
494536832,
494544512
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.1.attention_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
494544512,
494552192
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.1.feed_forward.w1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
494552192,
573195392
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"context_refiner.1.feed_forward.w2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
573195392,
651838592
],
"dtype": "BF16",
"shape": [
3840,
10240
]
},
"context_refiner.1.feed_forward.w3.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
651838592,
730481792
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"context_refiner.1.ffn_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
730481792,
730489472
],
"dtype": "BF16",
"shape": [
3840
]
},
"context_refiner.1.ffn_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
730489472,
730497152
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.0.adaLN_modulation.0.bias": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4262134912,
4262165632
],
"dtype": "BF16",
"shape": [
15360
]
},
"noise_refiner.0.adaLN_modulation.0.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4262165632,
4270029952
],
"dtype": "BF16",
"shape": [
15360,
256
]
},
"noise_refiner.0.attention.norm_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4270029952,
4270030208
],
"dtype": "BF16",
"shape": [
128
]
},
"noise_refiner.0.attention.norm_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4270030208,
4270030464
],
"dtype": "BF16",
"shape": [
128
]
},
"noise_refiner.0.attention.to_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4270030464,
4299521664
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.0.attention.to_out.0.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4299521664,
4329012864
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.0.attention.to_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4329012864,
4358504064
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.0.attention.to_v.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4358504064,
4387995264
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.0.attention_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4387995264,
4388002944
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.0.attention_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4388002944,
4388010624
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.0.feed_forward.w1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4388010624,
4466653824
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"noise_refiner.0.feed_forward.w2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4466653824,
4545297024
],
"dtype": "BF16",
"shape": [
3840,
10240
]
},
"noise_refiner.0.feed_forward.w3.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4545297024,
4623940224
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"noise_refiner.0.ffn_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4623940224,
4623947904
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.0.ffn_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4623947904,
4623955584
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.1.adaLN_modulation.0.bias": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4623955584,
4623986304
],
"dtype": "BF16",
"shape": [
15360
]
},
"noise_refiner.1.adaLN_modulation.0.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4623986304,
4631850624
],
"dtype": "BF16",
"shape": [
15360,
256
]
},
"noise_refiner.1.attention.norm_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4631850624,
4631850880
],
"dtype": "BF16",
"shape": [
128
]
},
"noise_refiner.1.attention.norm_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4631850880,
4631851136
],
"dtype": "BF16",
"shape": [
128
]
},
"noise_refiner.1.attention.to_k.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4631851136,
4661342336
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.1.attention.to_out.0.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4661342336,
4690833536
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.1.attention.to_q.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4690833536,
4720324736
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.1.attention.to_v.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4720324736,
4749815936
],
"dtype": "BF16",
"shape": [
3840,
3840
]
},
"noise_refiner.1.attention_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4749815936,
4749823616
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.1.attention_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4749823616,
4749831296
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.1.feed_forward.w1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4749831296,
4828474496
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"noise_refiner.1.feed_forward.w2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4828474496,
4907117696
],
"dtype": "BF16",
"shape": [
3840,
10240
]
},
"noise_refiner.1.feed_forward.w3.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4907117696,
4985760896
],
"dtype": "BF16",
"shape": [
10240,
3840
]
},
"noise_refiner.1.ffn_norm1.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4985760896,
4985768576
],
"dtype": "BF16",
"shape": [
3840
]
},
"noise_refiner.1.ffn_norm2.weight": {
"source_shard": "diffusion_pytorch_model-00001-of-00003.safetensors",
"source_data_offsets": [
4985768576,
4985776256
],
"dtype": "BF16",
"shape": [
3840
]
}
}