| from setuptools import setup, find_packages |
|
|
| setup( |
| name="codegeex", |
| py_modules=["codegeex"], |
| version="1.0", |
| description="CodeGeeX: A Open Multilingual Code Generation Model.", |
| author="Qinkai Zheng", |
| packages=find_packages(), |
| install_requires=[ |
| "fire>=0.4.0", |
| "ipython>=8.4.0", |
| "numpy>=1.22.0", |
| "pandas>=1.3.5", |
| "pyzmq>=23.2.1", |
| "regex>=2022.3.15", |
| "setuptools>=58.0.4", |
| "transformers>=4.22.0,<=4.24.0", |
| "tokenizers>=0.11.0,<=0.11.4", |
| "torch>=1.10.0", |
| "tqdm>=4.63.0", |
| "cpm_kernels", |
| "deepspeed>0.6.1", |
| ], |
| entry_points={} |
| ) |
|
|
