diff --git a/pyproject.toml b/pyproject.toml index fb066b8ed3..717f88f63a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ requires = [ "packaging>=24.2", "setuptools>=77.0.3,<80.0.0", "setuptools-scm>=8.0", - "torch==2.7.0", + "torch<=2.7.0", "wheel", "jinja2" ] diff --git a/requirements/common.txt b/requirements/common.txt index 61c547d7d1..b4f99acdab 100644 --- a/requirements/common.txt +++ b/requirements/common.txt @@ -1,15 +1,20 @@ -unsloth_zoo>=2025.6.2 -packaging +unsloth_zoo>=2025.6.7 +packaging>=24.1 tyro -transformers>=4.51.3,!=4.47.0,!=4.52.0,!=4.52.1,!=4.52.2 +transformers>=4.51.3,!=4.47.0,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3 datasets>=3.4.1 sentencepiece>=0.2.0 tqdm +tyro psutil wheel>=0.42.0 numpy accelerate>=0.34.1 trl>=0.7.9,!=0.9.0,!=0.9.1,!=0.9.2,!=0.9.3,!=0.15.0 peft>=0.7.1,!=0.11.0 +protobuf huggingface_hub hf_transfer +pillow +regex +msgspec diff --git a/setup.py b/setup.py index 0f4a86330e..4ea7ee7d4d 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):