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, ), ]