from setuptools import setup, find_packages

setup(
    name="finetunex",
    version="0.1.0",
    author="FineTuneX Team",
    description="大模型微调框架",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    python_requires=">=3.9",
    install_requires=[
        "torch>=2.0.0",
        "transformers>=4.40.0",
        "datasets>=2.14.0",
        "accelerate>=0.25.0",
        "peft>=0.7.0",
        "bitsandbytes>=0.41.0",
        "trl>=0.7.0",
        "fastapi>=0.104.0",
        "uvicorn>=0.24.0",
    ],
)