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