From 0a33947fb614ccfe33ee118da7d369d1d20c0cc3 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 30 Jun 2025 19:32:32 -0700 Subject: [PATCH] setup.py --- requirements/common.txt | 2 +- setup.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) 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(),