DiTAR: Diffusion Transformer Autoregressive Modeling for Speech Generation
Paper โข 2502.03930 โข Published
INT8 ConvRot conversion of FireRedTeam/FireRedTTS3 for
FireRedTTS3-ComfyUI, produced with the official
comfy-kitchen quantizer (TensorWiseINT8Layout.quantize, registry quantize_int8_convrot_weight).
Format per quantized Linear (current ComfyUI representation):
weight โ torch.int8, original [out, in] shape, contains the offline Hadamard-rotated weight (W @ H^T per 256-column group)weight_scale โ torch.float32, [out, 1] per-output-row scalebias โ original float biascomfy_quant โ uint8 JSON: {"format": "int8_tensorwise", "convrot": true, "convrot_groupsize": 256}At inference the companion custom node rotates activations online via
comfy_kitchen.int8_linear(..., convrot=True, convrot_groupsize=256) โ dynamic per-row INT8 activation
quantization + INT8 GEMM, rescaled by scale_x * scale_w. No whole-weight dequantization on the hot path.
| Component | Quantized | Kept float |
|---|---|---|
fireredtts3_base |
321/332 Linears (1.73B params, 81.5% of core): all backbone_llm.layers.*, patch_encoder.blocks.*, dit.blocks.* |
embeddings, norms, spk_proj_*, patch_encoder.in_proj/out_proj, dit_head, dit.in_proj (1600 % 256 != 0), dit.t_embedder, dit.final_layer, stop_head, Conv1d |
fireredtts3_instruct |
321/331 Linears (1.73B params, 71.2% of core): same block families (backbone_llm.model.layers.*) |
same exclusions |
redae |
nothing | everything |
campp |
nothing | everything |
| Core | Official fp32 | This repo |
|---|---|---|
fireredtts3_base |
8.48 GB | 3.30 GB |
fireredtts3_instruct |
8.48 GB | 3.30 GB |
redae / campp / tokenizer |
copied through unchanged |
comfy_kitchen.int8_linear runtime: worst rel-L2 0.01162, worst cosine 0.999932[N,1] and positive, Conv1d/RedAE/CAM++ untouchedConvRotInt8Linear modules, >42k counted INT8 ConvRot kernel calls during generation, weights stay int8 across unload/reload@article{fireredtts3,
title = {FireRedTTS3: Unified Speech Generation and Editing with Semantically Enriched Speech Representations},
author = {FireRed Team},
journal = {arXiv preprint},
year = {2026},
}
All credit to the FireRed Team โ see the upstream repo and model card. Apache-2.0.
Base model
FireRedTeam/FireRedTTS3