Upgrade torch, torchvision package versions

This commit is contained in:
Debanjum Singh Solanky 2022-08-20 14:45:43 +03:00
parent 972523e8a9
commit e6abe76875

View file

@ -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",