convert.py: add mapping for safetensors bf16

akx · akx · commit f519e81995c9 · 2023-05-26T13:02:05.000+03:00
Fixes ggml-org#1473
diff --git a/convert.py b/convert.py
@@ -739,6 +739,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,
