[Fix] update qwen config (#358)

764c2f79 · Leymore · GitHub · ab21f3be · 764c2f79 · 764c2f79
Unverified Commit 764c2f79 authored Sep 05, 2023 by Leymore Committed by GitHub Sep 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

configs/models/hf_qwen_7b.py configs/models/hf_qwen_7b.py +1 -0

configs/models/hf_qwen_7b_chat.py configs/models/hf_qwen_7b_chat.py +1 -0

No files found.
--- a/configs/models/hf_qwen_7b.py
+++ b/configs/models/hf_qwen_7b.py
@@ -19,6 +19,7 @@ models = [
            use_fast=False,
            revision='39fc5fdcb95c8c367bbdb3bfc0db71d96266de09'
        ),
+        pad_token_id=151643,
        max_out_len=100,
        max_seq_len=2048,
        batch_size=8,

--- a/configs/models/hf_qwen_7b_chat.py
+++ b/configs/models/hf_qwen_7b_chat.py
@@ -19,6 +19,7 @@ models = [
            truncation_side='left',
            trust_remote_code=True,
            use_fast=False,),
+        pad_token_id=151643,
        max_out_len=100,
        max_seq_len=2048,
        batch_size=8,