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

View File

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