File size: 616 Bytes
c1a46f7 6b73a07 c1a46f7 6b73a07 c1a46f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | 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",
],
)
|