diff --git a/requirements/common.txt b/requirements/common.txt index b4f99acdab..6fe0aab747 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -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 diff --git a/setup.py b/setup.py index 4ea7ee7d4d..469af43cdd 100644 --- a/setup.py +++ b/setup.py @@ -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(),