From 87b0a60cdd2a1dad71c5d60e60716dcfefa59e3f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sigbj=C3=B8rn=20Skj=C3=A6ret?= Date: Wed, 27 May 2026 15:06:18 +0200 Subject: [PATCH] pyproject : add conversion folder and update dependencies (#23746) * add conversion folder and update dependencies * limit python version for triton * update dev-dependencies section --- gguf-py/pyproject.toml | 2 +- pyproject.toml | 21 ++++++++++++--------- 2 files changed, 13 insertions(+), 10 deletions(-) diff --git a/gguf-py/pyproject.toml b/gguf-py/pyproject.toml index b2a36460a..d11c34a21 100644 --- a/gguf-py/pyproject.toml +++ b/gguf-py/pyproject.toml @@ -37,7 +37,7 @@ packages = [ [tool.poetry.dependencies] python = ">=3.10" -[tool.poetry.dev-dependencies] +[tool.poetry.group.dev.dependencies] pytest = "^5.2" [build-system] diff --git a/pyproject.toml b/pyproject.toml index a39c4f010..e4f8c86b9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,13 +6,13 @@ version = "0.0.0" dynamic = ["classifiers"] readme = "README.md" authors = [{name = "GGML", email = "ggml@ggml.ai"}] -requires-python = '>=3.10' +requires-python = '>=3.10,<3.15' dependencies = [ - 'numpy (>=1.25.0,<2.0.0)', + 'numpy (>=1.26.4,<3.0.0)', 'sentencepiece (>=0.1.98,<0.3.0)', 'transformers (==5.5.1)', - 'protobuf (>=4.21.0)', - 'torch (>=2.2.0,<3.0.0)', + 'protobuf (>=4.21.0,<5.0.0)', + 'torch (>=2.6.0,<3.0.0)', 'gguf @ ./gguf-py', ] classifiers = [ @@ -32,17 +32,20 @@ llama-convert-llama-ggml-to-gguf = "convert_llama_ggml_to_gguf:main" llama-ggml-vk-generate-shaders = "ggml_vk_generate_shaders:main" [tool.poetry] -packages = [{ include = "*.py", from = "." }] +packages = [ + { include = "*.py", from = "." }, + { include = "conversion", from = "." }, +] [tool.poetry.dependencies] torch = [ - { version = "~=2.6.0", source = "pypi", markers = "sys_platform == 'darwin'" }, - { version = "~=2.6.0+cpu", source = "pytorch", markers = "sys_platform == 'linux'" }, - { version = "~=2.6.0", source = "pypi", markers = "sys_platform == 'win32'" } + { version = "==2.11.0", source = "pypi", markers = "sys_platform == 'darwin'" }, + { version = "==2.11.0+cpu", source = "pytorch", markers = "sys_platform == 'linux'" }, + { version = "==2.11.0", source = "pypi", markers = "sys_platform == 'win32'" }, ] [tool.poetry.group.dev.dependencies] -pytest = "^5.2" +pytest = "~=8.3.3" # Force wheel + cpu # For discussion and context see https://github.com/python-poetry/poetry#6409