Upgrade torch, torchvision package versions

2026-03-02 21:19:12 +00:00 · 2022-08-20 14:45:43 +03:00
parent 972523e8a9
commit e6abe76875
1 changed files with 2 additions and 2 deletions
--- 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",