Uninstall torchvision and torchaudio for now

huggingface · Apr 15, 2024 · 0cb9fda · 0cb9fda
1 parent cbf021d
commit 0cb9fda
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -68,6 +68,7 @@ pypi_upload: ${PACKAGE_DIST} ${PACKAGE_WHEEL}
 
 # Tests
 test_installs:
+	python -m pip uninstall torchvision torchaudio
 	python -m pip install -r requirements.txt
 	python -m pip install .[tests]
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
 loguru == 0.6.0
 transformers == 4.39.3
-torch ~= 2.2.0
+torch ~= 2.2.0 --index-url https://download.pytorch.org/whl/cpu
 torch-xla[tpu] ~= 2.2.0 -f https://storage.googleapis.com/libtpu-releases/index.html