Patch unsloth-zoo llama_cpp.py and unsloth save.py from windows-support branch
This commit is contained in:
parent
e280e457d1
commit
c64e50b46f
1 changed files with 10 additions and 3 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue