feat: Add Cohere Command R / R+ model support (#3333)

This commit is contained in:
takatost 2024-04-11 01:22:55 +08:00 • committed by GitHub
commit 826c422ac4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
16 changed files with 404 additions and 128 deletions

View file

@ -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