feat: Add Cohere Command R / R+ model support (#3333)
This commit is contained in:
parent
bf63a43bda
commit
826c422ac4
16 changed files with 404 additions and 128 deletions
|
|
@ -232,8 +232,8 @@ class SimplePromptTransform(PromptTransform):
|
|||
)
|
||||
),
|
||||
max_token_limit=rest_tokens,
|
||||
ai_prefix=prompt_rules['human_prefix'] if 'human_prefix' in prompt_rules else 'Human',
|
||||
human_prefix=prompt_rules['assistant_prefix'] if 'assistant_prefix' in prompt_rules else 'Assistant'
|
||||
human_prefix=prompt_rules['human_prefix'] if 'human_prefix' in prompt_rules else 'Human',
|
||||
ai_prefix=prompt_rules['assistant_prefix'] if 'assistant_prefix' in prompt_rules else 'Assistant'
|
||||
)
|
||||
|
||||
# get prompt
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue