打开autodl,创建实例A100-PCIE-40GB * 1卡使用miniconda镜像。
方法一:从huggingface下载
这个步骤不稳定我们换种下载方式
apt-get update
apt install git-lfs
git lfs install
git clone https://huggingface.co/THUDM/chatglm2-6b
方法二:去清华云下载模型
这个链接只能下载所有的模型,包括chatglm和visualglm,我还没有找到只下载chatglm2的连接。
git clone https://github.com/chenyifanthu/THU-Cloud-Downloader.git
cd THU-Cloud-Downloader
pip install argparse requests tqdm
python main.py \
--link https://cloud.tsinghua.edu.cn/d/674208019e314311ab5c/\
--save ../autodl-tmp
这个下载过程要花半个小时,可以开个新的终端下载这个库并安装依赖
git clone https://github.com/THUDM/ChatGLM2-6B
cd ChatGLM2-6B
pip install -r requirements.txt
最后都下载好了,看哪个盘比较充裕就整合起来放在哪个盘。我放在了系统盘。
source /etc/network_turbo
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/THUDM/chatglm2-6b
mv model/* chatglm2-6b/
我们来测试一下:
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("../chatglm2-6b", trust_remote_code=True)
model = AutoModel.from_pretrained("../chatglm2-6b", trust_remote_code=True).half().cuda()
model = model.eval()
response, history = model.chat(tokenizer, "你好", history=[])
print(response)
成功啦!下一步是微调!
©版权声明
文章版权归作者所有,未经允许请勿转载。