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

conversation dataset (sharegpt format) custom template dev #767

Merged
merged 10 commits into from
Apr 16, 2024

Conversation

wheresmyhair
Copy link
Collaborator

@wheresmyhair wheresmyhair commented Apr 15, 2024

ready for review

@wheresmyhair wheresmyhair changed the title sharegpt custom template dev conversation dataset (sharegpt format) custom template dev Apr 15, 2024
@wheresmyhair wheresmyhair marked this pull request as ready for review April 16, 2024 03:27
Copy link
Contributor

@research4pan research4pan left a comment

Choose a reason for hiding this comment

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

LGTM, thanks! Two small questions that may help everyone use this feature easier:

  • I am wondering if there are examples of typical templates, ranked according to their popularity.
  • I am wondering how users can specify their own template in the data without modifying the code.

src/lmflow/models/hf_decoder_model.py

  • line 467-483: we can use mapping instead of a list of if-else, which is both more efficient and easier to maintain, but this can be improved later.

@research4pan research4pan merged commit a3d7f45 into main Apr 16, 2024
2 checks passed
@wheresmyhair wheresmyhair deleted the yizhenjia-sharegpt-template-dev branch April 17, 2024 08:00
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.

2 participants