from setuptools import setup, find_packages

setup(
    name="train",
    version="0.1.0",
    packages=find_packages(),
    scripts=[
        'bin/train'
    ],

    install_requires=[
        'accelerate>=0.25.0',
        'bitsandbytes>=0.45.0',
        'black>=22.0.0',
        'datasets>=2.14.6',
        'einops>=0.7.0',
        'flake8>=4.0.0',
        'ipykernel>=6.0.0', 
        'ipywidgets>=8.0.0',
        'peft>=0.8.0',
        'peft>=0.8.0',
        'pytest-cov>=4.0.0',
        'pytest>=7.0.0',
        'pyyaml>=6.0',
        'requests>=2.28.0',
        'sentencepiece>=0.1.99',
        'torch>=2.0.0',
        'transformers>=4.30.0',
        'trl>=0.7.8',
        'unsloth>=2025.3',
        'vllm>=0.8',
    ],
    python_requires='>=3.10',
)