From 0cb9fdaab32892007734791bb5155f0ba09bd128 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Mon, 15 Apr 2024 13:42:45 +0200
Subject: [PATCH] Uninstall torchvision and torchaudio for now

---
 Makefile         | 1 +
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 9fe7d51f..95127dd9 100644
--- 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
index 8332055f..16968f3f 100644
--- 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