| 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={} | |
| ) | |