from setuptools import setup, find_packages

setup(
    name="tynerox",
    version="0.1.0",
    description="TyneRox: custom rotary-transformer causal LM",
    author="Thiago Luiz Rodrigues",
    author_email="<EMAIL>",
    url="https://github.com/seu-usuario/tynerox",
    license="Apache-2.0",
    packages=find_packages("src"),
    package_dir={"": "src"},
    install_requires=[
        "torch>=2.6.0,<3.0.0",
        "transformers[torch]>=4.50.3,<5.0.0",
        "flash-attn>=2.7.4.post1,<3.0.0",
    ],
)