starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Wang Peng
2024-01-30 21:09:02 +08:00
committed by Ren Xuancheng
parent 99b13b4fd2
commit 1c34702e82
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -267,6 +267,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers import GenerationConfig from transformers import GenerationConfig
from qwen_generation_utils import make_context, decode_tokens, get_stop_words_ids from qwen_generation_utils import make_context, decode_tokens, get_stop_words_ids
# To generate attention masks automatically, it is necessary to assign distinct
# token_ids to pad_token and eos_token, and set pad_token_id in the generation_config.
tokenizer = AutoTokenizer.from_pretrained( tokenizer = AutoTokenizer.from_pretrained(
'./', './',
pad_token='<|extra_0|>', pad_token='<|extra_0|>',