dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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

Browse source
This commit is contained in:
Paul Gauthier 2024-07-10 15:29:45 +01:00 • committed by Paul Gauthier (aider)
parent f267f43094
commit 3414c3ab3b
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
setup.py
View file

@ -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