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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-19 08:02:44 +00:00
commit 7d69ca5ff0
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
studio/install_llama_prebuilt.py
View file

@ -2637,7 +2637,11 @@ def _pick_rocm_gfx_target(out: str) -> str | None:
_tokens = re.findall(r"gfx[1-9][0-9a-z]{2,3}", out.lower())
if not _tokens:
return None
_vis = os.environ.get("HIP_VISIBLE_DEVICES") or os.environ.get("ROCR_VISIBLE_DEVICES") or ""
_vis = (
os.environ.get("HIP_VISIBLE_DEVICES")
or os.environ.get("ROCR_VISIBLE_DEVICES")
or ""
)
if _vis:
_first = _vis.split(",")[0].strip()
try: