| 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", | |
| ], | |
| ) | |