Updated pytorch version requirement

Jacob Pennington · Jacob Pennington · commit 8d2b46d36c32 · 2025-07-16T18:26:30.000-04:00
diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
     'scipy',
     'scikit-learn',
     'tqdm',
-    'torch>=1.6',
+    'torch>=2.1.0',
     'numba',
     'faiss-cpu',
     'psutil',
