File size: 663 Bytes
594b008 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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={}
)
|