diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 5763b666..3b9fa264 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2482,6 +2482,10 @@ class BertModel(Model): print(f"Can not map tensor {name!r}") sys.exit() + # convert any unsupported data types to float32 + if data_torch.dtype not in (torch.float16, torch.float32): + data_torch = data_torch.to(torch.float32) + data = data_torch.squeeze().numpy() n_dims = len(data.shape) new_dtype: type[np.floating[Any]]