fixes

2026-10-18 07:33:20 +00:00 · 2024-12-25 21:34:56 +05:30
parent 5aef4a6c80
commit 6b54867c3c
4 changed files with 45 additions and 20 deletions
@@ -675,6 +675,7 @@ _DTYPE_MAP = {
    "fp16": torch.float16,
    "fp32": torch.float32,
 }
+_INVERSE_DTYPE_MAP = {v: k for k, v in _DTYPE_MAP.items()}


 def _map_to_args_type(args: Dict[str, Any]) -> Args: