diff --git a/install_python_stack.py b/install_python_stack.py index 7d3ab66b20..8f2b5fba8c 100644 --- a/install_python_stack.py +++ b/install_python_stack.py @@ -186,20 +186,27 @@ def install_python_stack() -> int: constrain=False, ) - # 6. Patch: override llama_cpp.py with fix from unsloth-zoo main branch + # 6. Patch: override llama_cpp.py with fix from unsloth-zoo feature/llama-cpp-windows-support branch patch_package_file( "unsloth-zoo", os.path.join("unsloth_zoo", "llama_cpp.py"), - "https://raw.githubusercontent.com/unslothai/unsloth-zoo/refs/heads/main/unsloth_zoo/llama_cpp.py", + "https://raw.githubusercontent.com/unslothai/unsloth-zoo/refs/heads/feature/llama-cpp-windows-support/unsloth_zoo/llama_cpp.py", ) - # 7. Patch: override vision.py with fix from unsloth PR #4091 + # 7a. Patch: override vision.py with fix from unsloth PR #4091 patch_package_file( "unsloth", os.path.join("unsloth", "models", "vision.py"), "https://raw.githubusercontent.com/unslothai/unsloth/80e0108a684c882965a02a8ed851e3473c1145ab/unsloth/models/vision.py", ) + # 7b. Patch : override save.py with fix from feature/llama-cpp-windows-support + patch_package_file( + "unsloth", + os.path.join("unsloth", "save.py"), + "https://raw.githubusercontent.com/unslothai/unsloth/refs/heads/feature/llama-cpp-windows-support/unsloth/save.py", + ) + # 8. Studio dependencies pip_install( "Installing studio dependencies",