Explorar o código

convert.py: add mapping for safetensors bf16 (#1598)

Fixes #1473
Aarni Koskela %!s(int64=2) %!d(string=hai) anos
pai
achega
3e08ae99ce
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      convert.py

+ 1 - 0
convert.py

@@ -828,6 +828,7 @@ def lazy_load_torch_file(outer_fp: IO[bytes], path: Path) -> ModelPlus:
 
 
 SAFETENSORS_DATA_TYPES: Dict[str, DataType] = {
+    'BF16': DT_BF16,
     'F16': DT_F16,
     'F32': DT_F32,
     'I32': DT_I32,