From: Aarni Koskela <redacted>
Date: Fri, 7 Jul 2023 13:12:49 +0000 (+0300)
Subject: convert.py: add mapping for safetensors bf16 (#1598)
X-Git-Tag: gguf-v0.4.0~505
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=3e08ae99ceb143d67f9273fda47541e9d98ff23f;p=pkg%2Fggml%2Fsources%2Fllama.cpp

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

Fixes #1473
---

diff --git a/convert.py b/convert.py
index 66509b99..7a2705e5 100644
--- a/convert.py
+++ b/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,