4ca24886a6
The explicit text_encoder_quant=nvfp4 path gated on the transformer smoke probe, which builds the dynamic-activation NVFP4 config, while the TE caster _cast_nvfp4 applies weight-only NVFP4WeightOnlyConfig. On a Blackwell build that carries the weight-only FP4 path but not the dynamic FP4 GEMM, the probe would fail and the encoder would silently stay dense even though the caster would run. Add a dedicated weight-only NVFP4 smoke probe (mirroring _cast_nvfp4's config) and route TE nvfp4 through it; int8 / fp8_dynamic keep the dynamic transformer probe since their TE casters are also dynamic-activation. |
||
|---|---|---|
| .. | ||
| assets | ||
| auth | ||
| core | ||
| hub | ||
| loggers | ||
| models | ||
| plugins | ||
| requirements | ||
| routes | ||
| state | ||
| storage | ||
| tests | ||
| utils | ||
| __init__.py | ||
| _platform_compat.py | ||
| cloudflare_tunnel.py | ||
| colab.py | ||
| main.py | ||
| run.py | ||
| startup_banner.py | ||