qwerrwe / setup.py
winglian's picture
docker layer caching, build w axolotl from base build
990bec6
raw
history blame
996 Bytes
from setuptools import setup, find_packages
install_requires = []
with open("./requirements.txt", "r") as requirements_file:
# don't include peft yet until we check the int4
# need to manually install peft for now...
reqs = [r.strip() for r in requirements_file.readlines() if "peft" not in r]
reqs = [r for r in reqs if r and r[0] != "#"]
for r in reqs:
install_requires.append(r)
setup(
name='axolotl',
version='0.1',
description="You know you're going to axolotl questions",
package_dir={'': 'src'},
packages=find_packages(),
install_requires=install_requires,
extras_require={
'int4': [
"alpaca_lora_4bit @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip",
],
'int4_triton': [
"alpaca_lora_4bit[triton] @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip",
],
'extras': [
'flash-attn',
'deepspeed',
]
},
)