| from setuptools import setup, find_packages | |
| setup( | |
| name="easytranslate", | |
| version="0.1.0", | |
| description="Transformer-based English-to-Chinese Translation Model", | |
| packages=find_packages(where="src"), | |
| package_dir={"": "src"}, | |
| python_requires=">=3.10", | |
| install_requires=[ | |
| "torch>=2.1.0", | |
| "transformers>=4.36.0", | |
| "datasets>=2.16.0", | |
| "tokenizers>=0.15.0", | |
| "sentencepiece>=0.1.99", | |
| "accelerate>=0.25.0", | |
| "sacrebleu>=2.4.0", | |
| "numpy>=1.24.0,<2.0", | |
| "omegaconf>=2.3.0", | |
| "rich>=13.0.0", | |
| "tqdm>=4.66.0", | |
| ], | |
| ) | |