Skip to content

Commit ef8f082

Browse files
committed
[Feature] enable unsloth on amd gpu
1 parent e2aafc7 commit ef8f082

File tree

8 files changed

+868
-598
lines changed

8 files changed

+868
-598
lines changed

pyproject.toml

Lines changed: 16 additions & 597 deletions
Large diffs are not rendered by default.

requirements/build.txt

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
cmake>=3.26
2+
ninja
3+
packaging>=24.2
4+
setuptools>=77.0.3,<80.0.0
5+
setuptools-scm>=8.0
6+
wheel
7+
jinja2

requirements/common.txt

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
unsloth_zoo>=2025.6.2
2+
packaging
3+
tyro
4+
transformers>=4.51.3,!=4.47.0,!=4.52.0,!=4.52.1,!=4.52.2
5+
datasets>=3.4.1
6+
sentencepiece>=0.2.0
7+
tqdm
8+
psutil
9+
wheel>=0.42.0
10+
numpy
11+
accelerate>=0.34.1
12+
trl>=0.7.9,!=0.9.0,!=0.9.1,!=0.9.2,!=0.9.3,!=0.15.0
13+
peft>=0.7.1,!=0.11.0
14+
huggingface_hub
15+
hf_transfer

requirements/cuda.txt

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# Common dependencies
2+
-r common.txt
3+
4+
torch==2.7.0
5+
torchaudio==2.7.0
6+
torchvision==0.22.0
7+
xformers==0.0.30

requirements/rocm.txt

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
# Common dependencies
2+
-r common.txt
3+
4+
--extra-index-url https://download.pytorch.org/whl/rocm6.4.1
5+
6+
torch==2.7.0
7+
torchvision==0.21.0
8+
torchaudio==2.7.0
9+
10+
triton==3.2
11+
cmake>=3.26,<4
12+
packaging
13+
setuptools>=77.0.3,<80.0.0
14+
setuptools-scm>=8
15+
wheel
16+
jinja2>=3.1.6
17+
amdsmi==6.4.1

setup.py

Lines changed: 804 additions & 0 deletions
Large diffs are not rendered by default.

unsloth/models/_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
__version__ = "2025.6.5"
15+
from unsloth.version import __version__
1616

1717
__all__ = [
1818
"SUPPORTS_BFLOAT16",

unsloth/version.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
__version__ = "2025.6.5"

0 commit comments

Comments
 (0)