Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ChatGLM2-6B_pytorch
Commits
ec44f3ff
Commit
ec44f3ff
authored
Nov 14, 2023
by
zhaoying1
Browse files
Update README.md
parent
a58f127e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
8 deletions
+12
-8
README.md
README.md
+12
-8
No files found.
README.md
View file @
ec44f3ff
...
...
@@ -40,13 +40,13 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk
docker run -dit --network=host --name=chatglm --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
docker exec -it chatglm /bin/bash
pip install transformers==4.28.0 -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
pip install accelerate sentencepiece mdtex2html gradio rouge_chinese nltk jieba datasets protobuf peft -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
pip install accelerate sentencepiece mdtex2html gradio rouge_chinese nltk jieba datasets protobuf peft
==0.5.0 pydantic==1.10.9
-i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
```
### Dockerfile(方式二)
```
docker build -t chatglm:latest .
docker run -dit --network=host --name=chatglm --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1
baichuan
:latest
docker exec -it chatglm /bin/bash
docker build -t chatglm
2
:latest .
docker run -dit --network=host --name=chatglm
2
--privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1
chatglm2
:latest
docker exec -it chatglm
2
/bin/bash
```
### Conda(方法三)
...
...
@@ -76,7 +76,7 @@ pip install -r requirements.txt
│ ├── dev.json
│ └── train.json
```
## 模型下载
##
#
模型下载
Hugging Face模型下载地址:
[
ChatGLM2-6B
](
https://huggingface.co/THUDM/chatglm2-6b
)
...
...
@@ -143,7 +143,11 @@ Hugging Face模型下载地址:
python cli_demo.py
程序会在命令行中进行交互式的对话,在命令行中输入指示并回车即可生成回复,输入 clear 可以清空对话历史,输入 stop 终止程序。
#### Results
## 精度
无
## Results
-
推理效果如下:
<div
align=
"center"
>
<img
src=
"./ptuning/media/cli.png"
width=
"650"
height=
"100"
>
...
...
@@ -155,11 +159,11 @@ Hugging Face模型下载地址:
### 算法类别
`
自然语言处理
`
`
对话问答
`
### 热点应用行业
`
nlp,智能聊天助手,科研
`
`
医疗,教育,科研,金融
`
## 源码仓库及问题反馈
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment