innovation64/CodeGeeX-test

1from setuptools import setup, find_packages2 3setup(4    name="codegeex",5    py_modules=["codegeex"],6    version="1.0",7    description="CodeGeeX: A Open Multilingual Code Generation Model.",8    author="Qinkai Zheng",9    packages=find_packages(),10    install_requires=[11        "fire>=0.4.0",12        "ipython>=8.4.0",13        "numpy>=1.22.0",14        "pandas>=1.3.5",15        "pyzmq>=23.2.1",16        "regex>=2022.3.15",17        "setuptools>=58.0.4",18        "transformers>=4.22.0,<=4.24.0",19        "tokenizers>=0.11.0,<=0.11.4",20        "torch>=1.10.0",21        "tqdm>=4.63.0",22        "cpm_kernels",23        "deepspeed>0.6.1",24    ],25    entry_points={}26)27