27 lines
645 B
Python
27 lines
645 B
Python
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",
|
|
"tokenizers>=0.11.0",
|
|
"torch>=1.10.0",
|
|
"tqdm>=4.63.0",
|
|
"cpm_kernels",
|
|
"deepspeed>0.6.1",
|
|
],
|
|
entry_points={}
|
|
)
|