diff --git a/unsloth/save.py b/unsloth/save.py index 940feb40f9..f8f884a9d3 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -853,9 +853,13 @@ def save_to_gguf( model_dtype = "f16" if model_dtype == "float16" else "bf16" # Convert quantization_method to list - quantization_method = \ - quantization_method if type(quantization_method) is list else list(quantization_method) - + if isinstance(quantization_method, list): pass + elif isinstance(quantization_method, str): quantization_method = [ quantization_method, ] + elif isinstance(quantization_method, tuple): quantization_method = list(quantization_method) + else: + raise TypeError("Unsloth: quantization_method can only be a string or a list of strings") + pass + # Check if bfloat16 is supported if model_dtype == "bf16" and not torch.cuda.is_bf16_supported(): logger.warning(