convert.py: add mapping for safetensors bf16

Fixes #1473
2023-05-26 13:02:05 +03:00 · 2023-05-26 13:02:05 +03:00 · f519e81995
commit f519e81995
parent ac7876ac20
1 changed files with 1 additions and 0 deletions
--- 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,