提供不同大模型的微调代码,实现在单GPU,多GPU上数据并行,模型并行的微调。
- sh install_env.sh
- 要注意torch的版本,cuda的版本,transformers的版本,详情见install_env.sh。因为我的测试环境是在容器里,可能还会有另外包需要安装,运行时缺啥就直接安装,核心就这几个。
chatglm-6b的微调
llama-7b的微调
基于DeepSpeed-Chat框架在bloom上的微调。
| Name | Name | Last commit date | ||
|---|---|---|---|---|
提供不同大模型的微调代码,实现在单GPU,多GPU上数据并行,模型并行的微调。
chatglm-6b的微调
llama-7b的微调
基于DeepSpeed-Chat框架在bloom上的微调。