-
Notifications
You must be signed in to change notification settings - Fork 4.6k
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
Support Index series models. #5910
Conversation
src/llamafactory/extras/constants.py
Outdated
DownloadSource.DEFAULT: "IndexTeam/Index-1.9B-Chat", | ||
DownloadSource.MODELSCOPE: "IndexTeam/Index-1.9B-Chat", | ||
}, | ||
"Index-1.9B-Character": { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
rename to Index-1.9B-Charater-Chat
for the auto chat template
if template is not None and any(suffix in name for suffix in ("-Chat", "-Instruct")): |
src/llamafactory/extras/constants.py
Outdated
DownloadSource.DEFAULT: "IndexTeam/Index-1.9B-Pure", | ||
DownloadSource.MODELSCOPE: "IndexTeam/Index-1.9B-Pure", | ||
}, | ||
"Index-1.9B-32K": { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
rename to Index-1.9B-Chat-32K
can you rebase this pr? |
eb6956c
to
a15a69a
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Sorry for the trouble, it's my first time working on a project of this size. |
@Cuiyn no problem, thanks for your contribution |
What does this PR do?
This PR add support for Index series models.
Before submitting
(I tested and verified the effect of the added code locally)