github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update README.md 

太让人迷惑立
pull/1288/head
曾小健 2023-06-27 11:36:56 +08:00 committed by GitHub
parent c9a7b2af80
commit a9002a162a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ptuning/README.md
View File

@ -138,7 +138,7 @@ from transformers import AutoConfig, AutoModel, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
``` ```
1. 如果需要加载的是新 Checkpoint（只包含 PrefixEncoder 参数）： 1. 如果需要加载的是新 Checkpoint（只需包含 PrefixEncoder 参数）：
```python ```python
config = AutoConfig.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, pre_seq_len=128) config = AutoConfig.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, pre_seq_len=128)