Never auto update pytorch (#2943)

chaiNNer-org · Jun 6, 2024 · 875b887 · 875b887
1 parent ee0b03f
commit 875b887
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/backend/src/packages/chaiNNer_pytorch/__init__.py b/backend/src/packages/chaiNNer_pytorch/__init__.py
@@ -31,14 +31,14 @@ def get_pytorch():
                 pypi_name="torch",
                 version="2.1.2",
                 size_estimate=55.8 * MB,
-                auto_update=True,
+                auto_update=False,
             ),
             Dependency(
                 display_name="TorchVision",
                 pypi_name="torchvision",
                 version="0.16.2",
                 size_estimate=1.3 * MB,
-                auto_update=True,
+                auto_update=False,
             ),
         ]
     else:
@@ -53,7 +53,7 @@ def get_pytorch():
                     if nvidia.is_available
                     else "https://download.pytorch.org/whl/cpu"
                 ),
-                auto_update=not nvidia.is_available,  # Too large to auto-update
+                auto_update=False,
             ),
             Dependency(
                 display_name="TorchVision",
@@ -65,7 +65,7 @@ def get_pytorch():
                     if nvidia.is_available
                     else "https://download.pytorch.org/whl/cpu"
                 ),
-                auto_update=not nvidia.is_available,  # Needs to match PyTorch version
+                auto_update=False,
             ),
         ]