diff --git a/utils/convert-helper-bitnet.py b/utils/convert-helper-bitnet.py index 5b4149ad..3e540da4 100644 --- a/utils/convert-helper-bitnet.py +++ b/utils/convert-helper-bitnet.py @@ -35,7 +35,8 @@ def main(): preprocess_script = utils_dir / "preprocess-huggingface-bitnet.py" convert_script = utils_dir / "convert-ms-to-gguf-bitnet.py" - llama_quantize_binary = project_root_dir / "build" / "bin" / "llama-quantize" + extension = ".exe" if sys.platform.startswith("win") else "" + llama_quantize_binary = project_root_dir / "build" / "bin" / f"llama-quantize{extension}" input_file = model_dir / "model.safetensors" input_backup_file = model_dir / "model.safetensors.backup" @@ -131,4 +132,4 @@ def main(): print(f"Warning: Could not remove backup {input_backup_file}: {e}") if __name__ == "__main__": - main() \ No newline at end of file + main()