| """Project-local native NVFP4 experiments for Fish Audio S2-Pro.""" | |
| from .checkpoint import load_mixed_nvfp4_checkpoint | |
| from .server import launch_mixed_nvfp4_thread_safe_queue | |
| from .modules import ( | |
| NVFP4FeedForward, | |
| NVFP4Linear, | |
| NVFP4TransformerBlock, | |
| apply_nvfp4_rounding_checkpoint, | |
| convert_s2_pro_nvfp4, | |
| load_direct_w4a4_m1_extension, | |
| load_rmsnorm_nvfp4_m1_extension, | |
| load_smallm_nvfp4_extension, | |
| load_silu_product_nvfp4_m1_extension, | |
| quantize_nvfp4, | |
| ) | |
| __all__ = [ | |
| "NVFP4FeedForward", | |
| "NVFP4Linear", | |
| "NVFP4TransformerBlock", | |
| "apply_nvfp4_rounding_checkpoint", | |
| "convert_s2_pro_nvfp4", | |
| "load_mixed_nvfp4_checkpoint", | |
| "launch_mixed_nvfp4_thread_safe_queue", | |
| "load_direct_w4a4_m1_extension", | |
| "load_rmsnorm_nvfp4_m1_extension", | |
| "load_smallm_nvfp4_extension", | |
| "load_silu_product_nvfp4_m1_extension", | |
| "quantize_nvfp4", | |
| ] | |