File size: 263 Bytes
fa2d87b | 1 2 3 4 5 6 7 8 9 10 | 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}")
|