Skip to content

GitLab

You need to sign in or sign up before continuing.
Unverified Commit 764c2f79 authored by Leymore's avatar Leymore Committed by GitHub
Browse files

[Fix] update qwen config (#358)

parent ab21f3be
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
configs/models/hf_qwen_7b.py
View file @ 764c2f79
...@@ -19,6 +19,7 @@ models = [ ...@@ -19,6 +19,7 @@ models = [
use_fast=False, use_fast=False,
revision='39fc5fdcb95c8c367bbdb3bfc0db71d96266de09' revision='39fc5fdcb95c8c367bbdb3bfc0db71d96266de09'
), ),
pad_token_id=151643,
max_out_len=100, max_out_len=100,
max_seq_len=2048, max_seq_len=2048,
batch_size=8, batch_size=8,
......
configs/models/hf_qwen_7b_chat.py
View file @ 764c2f79
...@@ -19,6 +19,7 @@ models = [ ...@@ -19,6 +19,7 @@ models = [
truncation_side='left', truncation_side='left',
trust_remote_code=True, trust_remote_code=True,
use_fast=False,), use_fast=False,),
pad_token_id=151643,
max_out_len=100, max_out_len=100,
max_seq_len=2048, max_seq_len=2048,
batch_size=8, batch_size=8,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment