From 9fa70b4b05e7d47fff34dcd425a3290b860183fc Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Tue, 9 Jul 2024 09:37:10 +0100
Subject: [PATCH] Install the cpu-only version of torch

---
 benchmark/Dockerfile | 6 +++---
 dev-requirements.txt | 4 +++-
 requirements.in      | 4 ++++
 requirements.txt     | 4 +++-
 4 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/benchmark/Dockerfile b/benchmark/Dockerfile
index aad17224f..85eede652 100644
--- a/benchmark/Dockerfile
+++ b/benchmark/Dockerfile
@@ -3,8 +3,8 @@ RUN apt-get update
 RUN apt-get install -y less git build-essential
 COPY requirements.txt /aider/requirements.txt
 COPY dev-requirements.txt /aider/dev-requirements.txt
-RUN pip install --upgrade pip
-RUN pip install -r /aider/requirements.txt
-RUN pip install -r /aider/dev-requirements.txt
+RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir -r /aider/requirements.txt
+RUN pip install --no-cache-dir -r /aider/dev-requirements.txt
 RUN git config --global --add safe.directory /aider
 WORKDIR /aider
diff --git a/dev-requirements.txt b/dev-requirements.txt
index 9e014122a..9c4e44fc3 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -1,9 +1,11 @@
 #
-# This file is autogenerated by pip-compile with Python 3.11
+# This file is autogenerated by pip-compile with Python 3.12
 # by the following command:
 #
 #    pip-compile --output-file=dev-requirements.txt dev-requirements.in
 #
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 alabaster==0.7.16
     # via sphinx
 babel==2.15.0
diff --git a/requirements.in b/requirements.in
index 92912c60d..4c2bde243 100644
--- a/requirements.in
+++ b/requirements.in
@@ -1,6 +1,10 @@
 #
 #    pip-compile requirements.in --upgrade
 #
+
+# Install the cpu-only version of torch
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 configargparse
 GitPython
 openai
diff --git a/requirements.txt b/requirements.txt
index b275cb61c..c4a0954ec 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,9 +1,11 @@
 #
-# This file is autogenerated by pip-compile with Python 3.11
+# This file is autogenerated by pip-compile with Python 3.12
 # by the following command:
 #
 #    pip-compile requirements.in
 #
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 aiohttp==3.9.5
     # via
     #   huggingface-hub