diff --git a/pyproject.toml b/pyproject.toml index 8895bf0686..e117fa79df 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [build-system] -requires = ["setuptools==80.9.0", "setuptools-scm==9.2.0"] +requires = ["setuptools==83.0.0", "setuptools-scm==10.2.1"] build-backend = "setuptools.build_meta" [project] @@ -90,9 +90,9 @@ studio = [ "matplotlib==3.10.9", "pandas", "nest_asyncio", - "datasets==4.3.0", + "datasets==5.0.0", "pyjwt", - "huggingface-hub==0.36.2", + "huggingface-hub==1.24.0", "structlog>=24.1.0", "diceware", "ddgs", @@ -101,8 +101,8 @@ studio = [ "httpx>=0.27.0", "fastmcp>=3.0.2", "sqlite-vec==0.1.9", - "pymupdf==1.27.2.3", - "pymupdf4llm==0.3.4", + "pymupdf==1.28.0", + "pymupdf4llm==1.28.0", "python-docx==1.2.0", ] @@ -121,14 +121,14 @@ huggingfacenotorch = [ "tyro", "protobuf", "sentencepiece>=0.2.0", - "datasets>=3.4.1,!=4.0.*,!=4.1.0,<4.4.0", + "datasets>=4.0,<5.1", "accelerate>=0.34.1", "peft>=0.18.0,!=0.11.0", "huggingface_hub>=0.34.0", "hf_transfer", "diffusers", - "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0", - "trl>=0.18.2,!=0.19.0,<=0.24.0", + "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.14.1", + "trl>=0.18.2,!=0.19.0,<=1.9.0", "sentence-transformers", ] # torchcodec backend for Gemma audio / datasets>=4 (#7225). @@ -139,13 +139,13 @@ huggingfacenotorch = [ # Gate on the platforms that have a wheel, matching # PLATFORM_LACKS_TORCHCODEC_WHEEL in studio/install_python_stack.py. audio-torch210 = [ - "torchcodec>=0.10.0,<0.11.0 ; python_version >= '3.10' and (((sys_platform == 'linux' or sys_platform == 'win32') and (platform_machine == 'x86_64' or platform_machine == 'AMD64')) or (sys_platform == 'darwin' and platform_machine == 'arm64'))", + "torchcodec>=0.10.0,<0.16.0 ; python_version >= '3.10' and (((sys_platform == 'linux' or sys_platform == 'win32') and (platform_machine == 'x86_64' or platform_machine == 'AMD64')) or (sys_platform == 'darwin' and platform_machine == 'arm64'))", ] audio-torch290 = [ - "torchcodec>=0.8.0,<0.10.0 ; python_version >= '3.10' and (((sys_platform == 'linux' or sys_platform == 'win32') and (platform_machine == 'x86_64' or platform_machine == 'AMD64')) or (sys_platform == 'darwin' and platform_machine == 'arm64'))", + "torchcodec>=0.8.0,<0.16.0 ; python_version >= '3.10' and (((sys_platform == 'linux' or sys_platform == 'win32') and (platform_machine == 'x86_64' or platform_machine == 'AMD64')) or (sys_platform == 'darwin' and platform_machine == 'arm64'))", ] audio-torch280 = [ - "torchcodec>=0.6.0,<0.8.0 ; python_version >= '3.9' and (((sys_platform == 'linux' or sys_platform == 'win32') and (platform_machine == 'x86_64' or platform_machine == 'AMD64')) or (sys_platform == 'darwin' and platform_machine == 'arm64'))", + "torchcodec>=0.6.0,<0.16.0 ; python_version >= '3.9' and (((sys_platform == 'linux' or sys_platform == 'win32') and (platform_machine == 'x86_64' or platform_machine == 'AMD64')) or (sys_platform == 'darwin' and platform_machine == 'arm64'))", ] huggingface = [ "unsloth[huggingfacenotorch]", @@ -640,8 +640,8 @@ colab-new = [ "unsloth_zoo>=2026.7.6", "packaging", "tyro", - "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0", - "datasets>=3.4.1,!=4.0.*,!=4.1.0,<4.4.0", + "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.14.1", + "datasets>=4.0,<5.1", "sentencepiece>=0.2.0", "tqdm", "psutil", @@ -656,7 +656,7 @@ colab-new = [ ] colab-no-deps = [ "accelerate>=0.34.1", - "trl>=0.18.2,!=0.19.0,<=0.24.0", + "trl>=0.18.2,!=0.19.0,<=1.9.0", "peft>=0.18.0", "xformers ; ('linux' in sys_platform or sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')", "bitsandbytes>=0.45.5,!=0.46.0,!=0.48.0",