File size: 512 Bytes
58d9159 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | 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",
],
)
|