From e6abe76875352b4f72d21e3406f285db2fc56665 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Sat, 20 Aug 2022 14:45:43 +0300 Subject: [PATCH] Upgrade torch, torchvision package versions --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index a8640aa3..bf9f3ce9 100644 --- a/setup.py +++ b/setup.py @@ -24,8 +24,8 @@ setup( ), install_requires=[ "numpy == 1.22.4", - "torch == 1.11.0", - "torchvision == 0.12.0", + "torch == 1.12.1", + "torchvision == 0.13.1", "transformers == 4.21.0", "sentence-transformers == 2.1.0", "openai == 0.20.0",