from __future__ import annotations def torch_dtype_name(precision: str) -> str | None: if precision in {"orbitquant", "fp32"}: return None if precision == "bf16": return "bfloat16" raise ValueError(f"unknown precision: {precision}")