diff --git a/extension/llm/export/gguf.py b/extension/llm/export/gguf.py index 259d935b882..5fefd22e3d8 100644 --- a/extension/llm/export/gguf.py +++ b/extension/llm/export/gguf.py @@ -380,6 +380,8 @@ def iter_gguf( yield tensor.name, flat.view(torch.float16).reshape(shape).to( torch.bfloat16 ) + elif tensor.tensor_type == GGMLQuantizationType.BF16: + yield tensor.name, flat.view(torch.bfloat16).reshape(shape).clone() else: raise ValueError(f"Unsupported GGUF quant type: {tensor.tensor_type}")