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