ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
unsloth/studio
History
Daniel Han cc6c8160d1 Add IS_ROCM flag to hardware module and fix AMD error message 
Hardware module changes:
- Add IS_ROCM global flag (bool, default False) to hardware.py
- detect_hardware() sets IS_ROCM=True when torch.version.hip is set,
  while keeping DeviceType as CUDA (torch.cuda.* works on ROCm via HIP)
- Print "ROCm (HIP X.Y)" instead of just "CUDA" when IS_ROCM is True
- get_package_versions() now returns a "rocm" key with the HIP version
  (torch.version.hip) alongside the existing "cuda" key
- Export IS_ROCM from studio/backend/utils/hardware/__init__.py

Tokenizer error message fix:
- Replace "We do not support AMD" with a helpful message pointing to
  ROCm installation docs at docs.unsloth.ai, since AMD is now supported
2026-03-31 08:34:05 +00:00
..
backend Add IS_ROCM flag to hardware module and fix AMD error message 2026-03-31 08:34:05 +00:00
frontend fix: skip download progress polling for exported GGUF models (#4709) 2026-03-31 07:21:23 +04:00
__init__.py Final cleanup 2026-03-12 18:28:04 +00:00
install_llama_prebuilt.py Add ROCm support to llama.cpp prebuilt installer 2026-03-31 08:34:05 +00:00
install_python_stack.py Add ROCm torch reinstall support to install_python_stack.py 2026-03-31 08:34:05 +00:00
LICENSE.AGPL-3.0 Add AGPL-3.0 license to studio folder 2026-03-09 19:36:25 +00:00
setup.bat Final cleanup 2026-03-12 18:28:04 +00:00
setup.ps1 Fix/bun windows bin detection (#4703) 2026-03-30 21:58:33 +04:00
setup.sh Fix/bun windows bin detection (#4703) 2026-03-30 21:58:33 +04:00
Unsloth_Studio_Colab.ipynb Allow install_python_stack to run on Colab (#4633) 2026-03-27 00:29:27 +04:00