ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
unsloth/studio/backend/utils/hardware
History
GoldenGrapeGentleman 450f5de507 Add IS_ROCM hardware flag and fix AMD error message 
Add IS_ROCM flag to hardware.py detect_hardware() (set when
torch.version.hip is present, DeviceType stays CUDA). Export IS_ROCM
from __init__.py. Add "rocm" key to get_package_versions().

Replace "We do not support AMD" error in tokenizer_utils.py with a
helpful message pointing to ROCm installation docs.

Co-authored-by: Daniel Han <danielhanchen@gmail.com>
2026-03-31 08:37:57 +00:00
..
__init__.py Add IS_ROCM hardware flag and fix AMD error message 2026-03-31 08:37:57 +00:00
hardware.py Add IS_ROCM hardware flag and fix AMD error message 2026-03-31 08:37:57 +00:00
nvidia.py [Studio] multi gpu finetuning/inference via "balanced_low0/sequential" device_map (#4602) 2026-03-30 02:33:15 -07:00
VRAM_ESTIMATION.md [Studio] multi gpu finetuning/inference via "balanced_low0/sequential" device_map (#4602) 2026-03-30 02:33:15 -07:00
vram_estimation.py [Studio] multi gpu finetuning/inference via "balanced_low0/sequential" device_map (#4602) 2026-03-30 02:33:15 -07:00