Added support for CPU-only PyTorch requirement in setup.py.

This commit is contained in:
Paul Gauthier 2024-07-10 15:29:45 +01:00 committed by Paul Gauthier (aider)
parent f267f43094
commit 3414c3ab3b

View file

@ -5,13 +5,10 @@ from setuptools import find_packages, setup
from aider import __version__ from aider import __version__
from aider.help_pats import exclude_website_pats from aider.help_pats import exclude_website_pats
def get_requirements(suffix=''):
# Find the torch requirement and replace it with the CPU only version, fname = "requirements" + suffix + '.txt'
# because the GPU versions are huge with open(fname) as f:
def get_requirements():
with open("requirements.txt") as f:
requirements = f.read().splitlines() requirements = f.read().splitlines()
return requirements return requirements