ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

setup.py

Browse source
This commit is contained in:
Daniel Han 2025-06-30 19:32:32 -07:00
commit 0a33947fb6
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
requirements/common.txt
View file

@ -1,5 +1,5 @@
unsloth_zoo>=2025.6.7
packaging>=24.1
packaging>=24.2
tyro
transformers>=4.51.3,!=4.47.0,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3
datasets>=3.4.1

6
setup.py
View file

@ -145,8 +145,8 @@ def get_requirements() -> list[str]:
return requirements
INSTINCT_ARCH = ("gfx942", "gfx90a")
RADEON_ARCH = ("gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201")
INSTINCT_ARCH = ("gfx942", "gfx90a",)
RADEON_ARCH = ("gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201",)
class RocmExtraInstallCommand(install):
@ -834,6 +834,8 @@ if _is_hip():
'install': RocmExtraInstallCommand
}
print(get_unsloth_version())
print(get_requirements())
setup(
# static metadata should rather go in pyproject.toml
version=get_unsloth_version(),