efficient_reasoning/setup.py at main · shuoli90/efficient_reasoning · 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
from setuptools import setup, find_packages


setup(
    name='efficient_reasoning',
    version="0.0.1",
    packages=find_packages(include=["efficient_reasoning", "efficient_reasoning.*"]),
    python_requires=">=3.10",
    install_requires=[
        "torch",
        "flash-attn",
        "transformers>=4.46.2",  # fixed the gradient accumulation bug, see: https://huggingface.co/blog/gradient_accumulation
        "datasets",
        "accelerate",
        "peft",
        "trl",
        "deepspeed",
        "vllm==0.8.3",
        "antlr4-python3-runtime",
        "numpy",
        "requests",
        "tqdm",
        "wandb",
    ],
    extras_require={
        "develop": [
            "black>=24.10.0",
            "flake8>=7.1.1",
            "mypy>=1.12.0",
        ],
    },
)