From 607586e0b7b26a4d4c5cf5e4830ac60eb2520540 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 26 Mar 2024 22:56:02 +0100 Subject: [PATCH] fix: downgrade torch (#1902) Signed-off-by: Ettore Di Giacinto --- .../python/common-env/transformers/transformers-nvidia.yml | 4 ++-- backend/python/common-env/transformers/transformers.yml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/python/common-env/transformers/transformers-nvidia.yml b/backend/python/common-env/transformers/transformers-nvidia.yml index 55361234..e8d8155b 100644 --- a/backend/python/common-env/transformers/transformers-nvidia.yml +++ b/backend/python/common-env/transformers/transformers-nvidia.yml @@ -89,8 +89,8 @@ dependencies: - six==1.16.0 - sympy==1.12 - tokenizers - - torch==2.2.1 - - torchvision==0.17.1 + - torch==2.1.2 + - torchvision==0.16.2 - torchaudio==2.1.2 - tqdm==4.66.1 - triton==2.1.0 diff --git a/backend/python/common-env/transformers/transformers.yml b/backend/python/common-env/transformers/transformers.yml index bdf8c36f..be378f67 100644 --- a/backend/python/common-env/transformers/transformers.yml +++ b/backend/python/common-env/transformers/transformers.yml @@ -76,8 +76,8 @@ dependencies: - six==1.16.0 - sympy==1.12 - tokenizers - - torch==2.2.1 - - torchvision==0.17.1 + - torch==2.1.2 + - torchvision==0.16.2 - torchaudio==2.1.2 - tqdm==4.66.1 - triton==2.1.0