sync precision-config/50x-fp4-f8-sample.json from ModelScope
Browse files
precision-config/50x-fp4-f8-sample.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_comment": "FLUX.2-klein 50x \u2014 Blackwell GPUs (SM120+: RTX 50 series + B100/B200/GB200/RTX PRO Blackwell). Attention + FFN FP4; modulation + embedders + head FP8.",
|
| 3 |
+
"transformer_blocks.attn": "f4",
|
| 4 |
+
"transformer_blocks.ff": "f4",
|
| 5 |
+
"transformer_blocks.ff_context": "f4",
|
| 6 |
+
"single_transformer_blocks.attn": "f4",
|
| 7 |
+
"double_stream_modulation_img": "f8",
|
| 8 |
+
"double_stream_modulation_txt": "f8",
|
| 9 |
+
"single_stream_modulation": "f8",
|
| 10 |
+
"x_embedder": "f8",
|
| 11 |
+
"context_embedder": "f8",
|
| 12 |
+
"norm_out": "f8",
|
| 13 |
+
"proj_out": "f8",
|
| 14 |
+
"time_guidance_embed": "f8"
|
| 15 |
+
}
|