Disabling GPU libraries

axioma-ai-labs · Jan 18, 2025 · 78ad50c · 78ad50c
1 parent 1612680
commit 78ad50c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/Pipfile b/Pipfile
@@ -49,7 +49,7 @@ pytest = "*"
 pytest-asyncio = "*"
 pytest-cov = "*"
 ruff = "*"
-torch = {version = "*", index = "https://download.pytorch.org/whl/cpu"}
+torch = {version = "*", index = "pypi", extras = ["cpu"]}
 transformers = {version = ">=4.38.0"}
 
 [requires]

diff --git a/Pipfile.lock b/Pipfile.lock