.

chaiNNer-org · Dec 29, 2023 · 28af796 · 28af796
1 parent c85763e
commit 28af796
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/src/packages/chaiNNer_pytorch/__init__.py b/backend/src/packages/chaiNNer_pytorch/__init__.py
@@ -47,14 +47,14 @@ def get_pytorch():
                 pypi_name="torch",
                 version="2.1.2+cu121" if nvidia_is_available else "2.1.2+cpu",
                 size_estimate=2 * GB if nvidia_is_available else 140 * MB,
-                extra_index_url=("https://download.pytorch.org/whl/cu121"),
+                extra_index_url="https://download.pytorch.org/whl/cu121",
             ),
             Dependency(
                 display_name="TorchVision",
                 pypi_name="torchvision",
                 version="0.16.2+cu121" if nvidia_is_available else "0.16.2+cpu",
                 size_estimate=2 * MB if nvidia_is_available else 800 * KB,
-                extra_index_url=("https://download.pytorch.org/whl/cu121"),
+                extra_index_url="https://download.pytorch.org/whl/cu121",
             ),
         ]