From 875b887b9d84477b4c821ccdb71ad3def4357ed1 Mon Sep 17 00:00:00 2001
From: Joey Ballentine <34788790+joeyballentine@users.noreply.github.com>
Date: Thu, 6 Jun 2024 09:01:32 -0400
Subject: [PATCH] Never auto update pytorch (#2943)

---
 backend/src/packages/chaiNNer_pytorch/__init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/backend/src/packages/chaiNNer_pytorch/__init__.py b/backend/src/packages/chaiNNer_pytorch/__init__.py
index a44852a9f..2c76dbb88 100644
--- 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,
             ),
         ]