diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 5763b6664e832..3b9fa264aa22c 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2482,6 +2482,10 @@ def write_tensors(self): 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]]