Reduce setup dependencies.

aimat-lab · Jan 5, 2024 · d864735 · d864735
1 parent a6f495f
commit d864735
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -18,9 +18,9 @@
         "keras>=3.0.0",
         # Backends
         "tf-nightly-cpu==2.16.0.dev20231103",
-        "torch>=2.1.0",
-        "torchvision>=0.16.0",
-        "jax[cpu]",
+        # "torch>=2.1.0",
+        # "torchvision>=0.16.0",
+        # "jax[cpu]",
         # "torchrec",
         "numpy>=1.23.0",
         "scikit-learn>=1.1.3",