diff --git a/pyproject.toml b/pyproject.toml index ea86bfd20e..9dd8b41fc6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,8 +1,6 @@ [build-system] # Should be mirrored in requirements/build.txt requires = [ - "cmake>=3.26", - "ninja", "packaging>=24.2", "setuptools>=77.0.3,<80.0.0", "setuptools-scm>=8.0", @@ -13,7 +11,7 @@ build-backend = "setuptools.build_meta" name = "unsloth" description = "2-5X faster LLM finetuning" readme = "README.md" -requires-python = ">=3.9,<3.13" +requires-python = ">=3.9" dynamic = ["version", "dependencies", "optional-dependencies"] license = { file = "LICENSE" } keywords = ["ai", "llm",] diff --git a/setup.py b/setup.py index e75a10d409..703604c54a 100644 --- a/setup.py +++ b/setup.py @@ -770,6 +770,10 @@ if HAS_HIP: os.mkdir('thirdparties') os.chdir('thirdparties') + # Cmake and ninja + subprocess.check_call(['pip', 'install', 'cmake>=3.26']) + subprocess.check_call(['pip', 'install', 'ninja']) + # Extract ROCm GPU arch from environment variable. If unset, then detect ROCm arch from rocminfo # Refer to https://github.com/bitsandbytes-foundation/bitsandbytes/blob/1abd5e781013a085f86586b30a248dc769909668/bitsandbytes/cuda_specs.py#L81 # TODO(billishyahao): need to triage rocminfo unavailable observation from https://github.com/bitsandbytes-foundation/bitsandbytes/issues/1444