File size: 996 Bytes
77fca25
 
 
 
 
4131183
77fca25
f50de1b
77fca25
 
 
 
 
 
 
 
 
 
 
4131183
 
77fca25
 
4131183
77fca25
990bec6
 
 
 
77fca25
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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',
        ]
    },
)