From de6d02e17d66b4730cf84446b39c3388beca6908 Mon Sep 17 00:00:00 2001 From: Abukhoyer Shaik Date: Mon, 28 Jul 2025 05:36:39 +0000 Subject: [PATCH 1/3] torch upgrade to 2.7.0 and torchvision to 0.22.0 Signed-off-by: Abukhoyer Shaik --- pyproject.toml | 6 +++--- scripts/Jenkinsfile | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 479736c22..218fd84b8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,11 +39,11 @@ dependencies = [ "fire", "py7zr", "torchmetrics==1.7.0", - "torch==2.4.1; platform_machine=='aarch64'", + "torch==2.7.0; platform_machine=='aarch64'", # Specifying torch cpu package URL per python version, update the list once pytorch releases whl for python>3.11 "torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp38-cp38-linux_x86_64.whl ; python_version=='3.8' and platform_machine=='x86_64'", - "torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp39-cp39-linux_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'", - "torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp310-cp310-linux_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'", + "torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp39-cp39-manylinux_2_28_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'", + "torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'", ] [project.optional-dependencies] diff --git a/scripts/Jenkinsfile b/scripts/Jenkinsfile index 103c04b73..8d5bcc72d 100644 --- a/scripts/Jenkinsfile +++ b/scripts/Jenkinsfile @@ -24,7 +24,7 @@ pipeline { pip install .[test] && pip install junitparser pytest-xdist && pip install librosa==0.10.2 soundfile==0.13.1 && #packages needed to load example for whisper testing - pip install --extra-index-url https://download.pytorch.org/whl/cpu timm==1.0.14 torchvision==0.19.1+cpu einops==0.8.1 && #packages to load VLMs + pip install --extra-index-url https://download.pytorch.org/whl/cpu timm==1.0.14 torchvision==0.22.0+cpu einops==0.8.1 && #packages to load VLMs pip install /opt/qti-aic/integrations/torch_qaic/py310/torch_qaic-0.1.0-cp310-cp310-linux_x86_64.whl && # For finetuning tests rm -rf QEfficient" ''' From 48d7089c01808c35ba71a1c6ada6d1e851f5e2e8 Mon Sep 17 00:00:00 2001 From: Abukhoyer Shaik Date: Thu, 7 Aug 2025 10:41:49 +0000 Subject: [PATCH 2/3] upgrading to 2.7.1 Signed-off-by: Abukhoyer Shaik --- pyproject.toml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 218fd84b8..ddf663914 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,11 +39,11 @@ dependencies = [ "fire", "py7zr", "torchmetrics==1.7.0", - "torch==2.7.0; platform_machine=='aarch64'", + "torch==2.7.1; platform_machine=='aarch64'", # Specifying torch cpu package URL per python version, update the list once pytorch releases whl for python>3.11 "torch@https://download.pytorch.org/whl/cpu/torch-2.4.1%2Bcpu-cp38-cp38-linux_x86_64.whl ; python_version=='3.8' and platform_machine=='x86_64'", - "torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp39-cp39-manylinux_2_28_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'", - "torch@https://download.pytorch.org/whl/cpu/torch-2.7.0%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'", + "torch@https://download.pytorch.org/whl/cpu/torch-2.7.1%2Bcpu-cp39-cp39-manylinux_2_28_x86_64.whl ; python_version=='3.9' and platform_machine=='x86_64'", + "torch@https://download.pytorch.org/whl/cpu/torch-2.7.1%2Bcpu-cp310-cp310-manylinux_2_28_x86_64.whl ; python_version=='3.10' and platform_machine=='x86_64'", ] [project.optional-dependencies] From 4aeb14b236d8f0e5a2f050da4ce9a4553db6df81 Mon Sep 17 00:00:00 2001 From: Abukhoyer Shaik Date: Tue, 12 Aug 2025 05:34:34 +0000 Subject: [PATCH 3/3] upgrading torchvision to 0.22.1 for torch2.7.1 compatible Signed-off-by: Abukhoyer Shaik --- scripts/Jenkinsfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/Jenkinsfile b/scripts/Jenkinsfile index 230892f61..4eaf7883e 100644 --- a/scripts/Jenkinsfile +++ b/scripts/Jenkinsfile @@ -24,7 +24,7 @@ pipeline { pip install .[test] && pip install junitparser pytest-xdist && pip install librosa==0.10.2 soundfile==0.13.1 && #packages needed to load example for whisper testing - pip install --extra-index-url https://download.pytorch.org/whl/cpu timm==1.0.14 torchvision==0.22.0+cpu einops==0.8.1 && #packages to load VLMs + pip install --extra-index-url https://download.pytorch.org/whl/cpu timm==1.0.14 torchvision==0.22.1+cpu einops==0.8.1 && #packages to load VLMs pip install /opt/qti-aic/integrations/torch_qaic/py310/torch_qaic-0.1.0-cp310-cp310-linux_x86_64.whl && # For finetuning tests rm -rf QEfficient" '''