Make cupy a dev dependency

EleutherAI · Mar 17, 2023 · 89ca7f7 · 89ca7f7
1 parent acea481
commit 89ca7f7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,6 +37,7 @@ version = "0.1.1"
 
 [project.optional-dependencies]
 dev = [
+    "cupy-cuda11x",
     "hypothesis",
     "pre-commit",
     "pytest",