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-03-31 08:31:55 +00:00
commit 8df5d7d176
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/backend/utils/hardware/hardware.py
View file

@ -103,7 +103,9 @@ def detect_hardware() -> DeviceType:
# DeviceType stays CUDA since torch.cuda.* works on ROCm via HIP.
if getattr(torch.version, "hip", None) is not None:
IS_ROCM = True
print(f"Hardware detected: ROCm (HIP {torch.version.hip}) -- {device_name}")
print(
f"Hardware detected: ROCm (HIP {torch.version.hip}) -- {device_name}"
)
else:
print(f"Hardware detected: CUDA -- {device_name}")
return DEVICE