CloneUs/setup.py at master · Rypo/CloneUs · GitHub

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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from setuptools import setup, find_packages

setup(
    name = 'cloneus',
    version='0.0.2',
    author='Rypo',
    keywords="LLM, multi-user RP finetuning, discord, AI, chatbot",
    package_dir={"": "src"},
    packages = find_packages(where="src"),
    python_requires=">=3.10",
    install_requires=[
        'python-dotenv',
        'omegaconf',
        'more-itertools',
        'unidecode',
        'orjson',
        'ipython',
        'matplotlib',
        'google-api-python-client',
        'pandas',
        'bitsandbytes',
        'transformers',
        'datasets',
        'peft',
        'accelerate',
        'optimum',
        'safetensors',
        'trl',
        'wandb',
        'flash-attn',
        'unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git',
    ],
    extras_require={
        'discord': [
            'discord.py',
            # for image generation
            'diffusers', # do not use diffusers[torch] or it will try to install torch/triton=2.1.2
            'DeepCache'
        ],
        'quants': [ # this is fine with pytorch 2.2.0
            'auto-gptq',
            'autoawq',
            'exllamav2',
            'hqq',
            'aqlm[gpu,cpu]'
        ],
    },
    url='https://github.com/Rypo/CloneUs',
    license='MIT',
)