Commit a6f016b9 authored by zhouxiang's avatar zhouxiang
Browse files

增加镜像获取方式

parent f4b19218
...@@ -16,7 +16,7 @@ GLM是一个基于transformer的语言模型,利用自回归空白填充作为 ...@@ -16,7 +16,7 @@ GLM是一个基于transformer的语言模型,利用自回归空白填充作为
在光源可拉取推理的docker镜像,拉取方式如下: 在光源可拉取推理的docker镜像,拉取方式如下:
``` ```
docker pull docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:glm-ft-v1.0
``` ```
### 编译方法 ### 编译方法
...@@ -65,6 +65,14 @@ python ../examples/cpp/glm/glm_weight_convt.py -i /home/glm-130b-sat/49300/ -o ...@@ -65,6 +65,14 @@ python ../examples/cpp/glm/glm_weight_convt.py -i /home/glm-130b-sat/49300/ -o
mpirun -n 8 --allow-run-as-root ./bin/glm_example mpirun -n 8 --allow-run-as-root ./bin/glm_example
``` ```
此example程序会读取../examples/cpp/glm/start_ids.csv文件中的id作为输入token,生成的结果tokenid会保存在./out内,可以执行如下命令进行解析out结果:
```
python ../examples/pytorch/glm/glm_tokenize.py
```
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
[ModelZoo / GLM130B_FT · GitLab (hpccube.com)](https://developer.hpccube.com/codes/modelzoo/glm130b_ft) [ModelZoo / GLM130B_FT · GitLab (hpccube.com)](https://developer.hpccube.com/codes/modelzoo/glm130b_ft)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment