Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update baichuan chat template #2185

Merged
merged 4 commits into from
Aug 9, 2023
Merged

Conversation

Tomorrowxxy
Copy link
Contributor

Why are these changes needed?

Update baichuan-13b-chat tokenizer rules

Related issue number (if applicable)

#2175

Checks

  • I've run format.sh to lint the changes in this PR.
  • I've included any doc changes needed.
  • I've made sure the relevant tests are passing (if applicable).

fastchat/conversation.py Outdated Show resolved Hide resolved
sep="",
sep2="</s>",
stop_token_ids=[2, 195],
sep2="",
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
sep2="",

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fastchat/conversation.py Outdated Show resolved Hide resolved
stop_token_ids=[2, 195],
stop_token_ids=[],
Copy link
Member

@merrymercy merrymercy Aug 8, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could you clarify why the stop token IDs are being deleted? Are they no longer required?
Did you test the CLI and check whether the model can stop successfully?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I tested the submitted code in vllm and everything is fine and the inference results are as expected
vllm uses fschat

@merrymercy merrymercy changed the title update baichuan StarChat template update baichuan chat template Aug 8, 2023
@merrymercy merrymercy merged commit 2689353 into lm-sys:main Aug 9, 2023
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants