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

Support CohereForAI/c4ai-command-r-v01 #480

Closed
choyakawa opened this issue Mar 14, 2024 · 1 comment
Closed

Support CohereForAI/c4ai-command-r-v01 #480

choyakawa opened this issue Mar 14, 2024 · 1 comment

Comments

@choyakawa
Copy link

It is almost llama, except: logits_scale and different DecoderLayer

see Lightning-AI/litgpt#1089 ggerganov/llama.cpp#6033

@LZHgrla
Copy link
Collaborator

LZHgrla commented Mar 15, 2024

@choyakawa Hi

In my opinion, you can directly use CohereForAI/c4ai-command-r-v01 in xtuner.

Just like the modification on #429

  1. Apply CohereForAI/c4ai-command-r-v01 in your config, that is pretrained_model_name_or_path = 'CohereForAI/c4ai-command-r-v01'

  2. Expand the PROMPT_TEMPLATE according to the model and use it in your config, that is prompt_template = PROMPT_TEMPLATE.xxx

You can try it and if you encounter any problems, feel free to post them on this issue.

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

No branches or pull requests

2 participants