模型下载
大家需要从huggingface下载CodeGeex2的模型文件,可以直接下载到本地,也可以git clone,这里给huggingface的地址,大家自行下载
THUDM/codegeex2-6b · Hugging Face
模型部署
首先我们得有个环境去跑起来,这里我用的是阿里云的平台,具体怎么申请,可以看我的第一篇博客。首先新建一个文件夹codeGeex2,然后继续建一个codegeex2-6b的文件夹,我们要把下载的模型相关文件全部放在这个文件夹里面。我的如图所示
模型测试
这里官方给了测试代码,github地址如下:GitHub - THUDM/CodeGeeX2: CodeGeeX2: A More Powerful Multilingual Code Generation Model我们只需要把python测试代码的模型路径改成自己的地址就行,然后就可以测试啦