diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-07-07 16:12:49 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-07 09:12:49 -0400 |
| commit | 3e08ae99ceb143d67f9273fda47541e9d98ff23f (patch) | |
| tree | b69cc2fbd51bd70eee505a19a0fc0d76ad2537ef /convert.py | |
| parent | 481f793acc3882a09d45d8d2c3076ad3d1c60cfc (diff) | |
convert.py: add mapping for safetensors bf16 (#1598)
Fixes #1473
Diffstat (limited to 'convert.py')
| -rw-r--r-- | convert.py | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -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, |