From 3414c3ab3b688f6fc403e45abd5450d26bc399ce Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 10 Jul 2024 15:29:45 +0100 Subject: [PATCH] Added support for CPU-only PyTorch requirement in setup.py. --- setup.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/setup.py b/setup.py index e4d9b0d39..7cec0190a 100644 --- a/setup.py +++ b/setup.py @@ -5,13 +5,10 @@ from setuptools import find_packages, setup from aider import __version__ from aider.help_pats import exclude_website_pats - -# Find the torch requirement and replace it with the CPU only version, -# because the GPU versions are huge -def get_requirements(): - with open("requirements.txt") as f: +def get_requirements(suffix=''): + fname = "requirements" + suffix + '.txt' + with open(fname) as f: requirements = f.read().splitlines() - return requirements