ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix Windows AMD: route has_rocm hosts to HIP prebuilt path

Browse source 
resolve_release_asset_choice was selecting windows-cpu for all Windows
x86_64 hosts including those with has_rocm=True. Windows AMD users
should fall through to resolve_upstream_asset_choice which tries the
HIP prebuilt first. Add "not host.has_rocm" guard to the published
windows-cpu selection.
This commit is contained in:
Daniel Han 2026-04-05 03:31:04 +00:00
commit c6f5b3af32
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
studio/install_llama_prebuilt.py
View file

@ -3120,7 +3120,7 @@ def resolve_release_asset_choice(
)
published_choice: AssetChoice | None = None
if host.is_windows and host.is_x86_64:
if host.is_windows and host.is_x86_64 and not host.has_rocm:
published_choice = published_asset_choice_for_kind(release, "windows-cpu")
elif host.is_macos and host.is_arm64:
published_choice = published_asset_choice_for_kind(release, "macos-arm64")