torch requirements

kyegomez · Jul 2, 2023 · cf154a3 · cf154a3
1 parent 865371a
commit cf154a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -16,7 +16,7 @@
     'transformers'
   ],
   install_requires=[
-    "torch -f https://download.pytorch.org/whl/nightly/cu118/torch_nightly.html",
+    "torch"
     "lion-pytorch",
     "numpy",
     # colt5_attention