Fix setup.py

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"
 ]

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

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):