mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-25 23:05:00 +00:00
Added support for CPU-only PyTorch requirement in setup.py.
This commit is contained in:
parent
f267f43094
commit
3414c3ab3b
1 changed files with 3 additions and 6 deletions
9
setup.py
9
setup.py
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue