Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

ROCM support#3279

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to ourterms of service andprivacy statement. We’ll occasionally send you account related emails.

Already on GitHub?Sign in to your account

Open
electron271 wants to merge17 commits intounslothai:main
base:main
Choose a base branch
Loading
fromelectron271:main
Open
Changes fromall commits
Commits
Show all changes
17 commits
Select commitHold shift + click to select a range
4f519cd
add rocm support
electron271Sep 5, 2025
3c7155c
switch to bitsandbytes preview
electron271Sep 5, 2025
805db92
fix wrong version
electron271Sep 5, 2025
66a313c
switch to my own bitsandbytes url
electron271Sep 5, 2025
ba87680
change bitsandbytes url
electron271Sep 5, 2025
b5790fb
update bitsandbytes url
electron271Sep 6, 2025
dae212f
add torch wheel urls
electron271Sep 6, 2025
7d34d10
fix toml mistake
electron271Sep 6, 2025
0392028
remove xformers due to issues
electron271Sep 6, 2025
7160198
change bitsandbytes url
electron271Sep 6, 2025
c390dd1
update pyproject.toml to follow cuda name convention for rocm
electron271Sep 10, 2025
6ac2b01
Merge branch 'unslothai:main' into main
electron271Oct 3, 2025
d602036
Merge branch 'main' into main
electron271Oct 24, 2025
b793bda
Merge branch 'main' into main
electron271Oct 28, 2025
8d0f495
fix bitsandbytes url
electron271Dec 1, 2025
cf62269
Merge branch 'unslothai:main' into main
electron271Dec 1, 2025
3ac0abf
Merge branch 'unslothai:main' into main
electron271Dec 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 19 additions & 0 deletionspyproject.toml
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -958,6 +958,25 @@ amd = [
"bitsandbytes @ https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_main/bitsandbytes-1.33.7.preview-py3-none-win_amd64.whl ; (sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
"bitsandbytes @ https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_main/bitsandbytes-1.33.7.preview-py3-none-manylinux_2_24_aarch64.whl ; ('linux' in sys_platform) and (platform_machine == 'aarch64')",
]
rocm64-torch280 = [
"packaging",
"pytorch-triton-rocm",
"ninja",
"unsloth[huggingface]",
"bitsandbytes @ https://electron271.github.io/bitsandbytes-index/permalinks/c7d8ee9ed980191b/bitsandbytes-0.49.0.dev0-py3-none-manylinux_2_24_x86_64.whl#sha256=c7d8ee9ed980191b21255e0e34469c11db80a9ad28a078d66c3f02c228aa1f6c ; platform_system == 'Linux' and python_version >= '3.9' and platform_machine == 'x86_64'",

"pytorch_triton_rocm @ https://download.pytorch.org/whl/pytorch_triton_rocm-3.4.0-cp39-cp39-linux_x86_64.whl#sha256=7f26455680a068b04456c6977348d404ff31d2fef1d7472e980483db7acbe462 ; platform_system == 'Linux' and python_version == '3.9' and platform_machine == 'x86_64'",
"pytorch_triton_rocm @ https://download.pytorch.org/whl/pytorch_triton_rocm-3.4.0-cp310-cp310-linux_x86_64.whl#sha256=1ee0a5cf569175e63b43bc334dcaaf6f9b0d88eb455a452869c2bab14e1f7eb4 ; platform_system == 'Linux' and python_version == '3.10' and platform_machine == 'x86_64'",
"pytorch_triton_rocm @ https://download.pytorch.org/whl/pytorch_triton_rocm-3.4.0-cp311-cp311-linux_x86_64.whl#sha256=b0362725d8e16d185251e3dcd48455ebf9cdaad2c26052bb47ef08a1d687ed20 ; platform_system == 'Linux' and python_version == '3.11' and platform_machine == 'x86_64'",
"pytorch_triton_rocm @ https://download.pytorch.org/whl/pytorch_triton_rocm-3.4.0-cp312-cp312-linux_x86_64.whl#sha256=7afe951b9fc38f1a5b3a7b98bebbaa092bf51e6192b699b4fade9b1ad6fc9c2c ; platform_system == 'Linux' and python_version == '3.12' and platform_machine == 'x86_64'",
"pytorch_triton_rocm @ https://download.pytorch.org/whl/pytorch_triton_rocm-3.4.0-cp313-cp313-linux_x86_64.whl#sha256=1e7ccba3501fcd38e8cd8415f97a654043370e1fdc5a936bb75abe1bebeb94c9 ; platform_system == 'Linux' and python_version == '3.13' and platform_machine == 'x86_64'",

"torch @ https://download.pytorch.org/whl/rocm6.4/torch-2.8.0%2Brocm6.4-cp39-cp39-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.9' and platform_machine == 'x86_64'",
"torch @ https://download.pytorch.org/whl/rocm6.4/torch-2.8.0%2Brocm6.4-cp310-cp310-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.10' and platform_machine == 'x86_64'",
"torch @ https://download.pytorch.org/whl/rocm6.4/torch-2.8.0%2Brocm6.4-cp311-cp311-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.11' and platform_machine == 'x86_64'",
"torch @ https://download.pytorch.org/whl/rocm6.4/torch-2.8.0%2Brocm6.4-cp312-cp312-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.12' and platform_machine == 'x86_64'",
"torch @ https://download.pytorch.org/whl/rocm6.4/torch-2.8.0%2Brocm6.4-cp313-cp313-manylinux_2_28_x86_64.whl ; platform_system == 'Linux' and python_version == '3.13' and platform_machine == 'x86_64'",
]

[project.urls]
homepage = "http://www.unsloth.ai"
Expand Down
[8]ページ先頭
©2009-2025 Movatter.jp