From 3473ca61e75d14fe0edafc4c42b0baecbe7a0765 Mon Sep 17 00:00:00 2001 From: Gaspar Rochette Date: Tue, 16 Dec 2025 14:44:13 +0000 Subject: [PATCH] build: unpin torch version --- pyproject.toml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index bc2ca641..e1377da7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -97,8 +97,8 @@ classifiers = [ "Programming Language :: Python :: 3.12", ] dependencies = [ - "torch==2.7.0", - "torchvision==0.22.0", + "torch>=2.7.0", + "torchvision>=0.22.0", "torchmetrics[image]==1.7.4", "requests>=2.31.0", "transformers", @@ -128,7 +128,7 @@ dependencies = [ "ctranslate2==4.6.0", "whisper-s2t==1.3.1", "hqq==0.2.7.post1", - "torchao==0.12.0", #ref: https://github.com/pytorch/ao/issues/2919#issue-3375688762 + "torchao>=0.12.0", #ref: https://github.com/pytorch/ao/issues/2919#issue-3375688762 "llmcompressor", "gliner; python_version >= '3.10'", "piq", @@ -142,7 +142,7 @@ dependencies = [ [project.optional-dependencies] stable-fast = [ - "xformers==0.0.30", + "xformers>=0.0.30", "stable-fast-pruna==1.0.7", ] # dependencies are added here because the wheels aren't bundling them @@ -156,7 +156,7 @@ gptq = [ "gptqmodel; sys_platform == 'darwin' and platform_machine == 'arm64'", ] full = [ - "xformers==0.0.30", + "xformers>=0.0.30", "stable-fast-pruna==1.0.7", ] dev = [