Skip to content

Commit 0b09cee

Browse files
authored
Version names for Command R/R+ (#3491)
1 parent 36f7807 commit 0b09cee

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

fastchat/model/model_adapter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2423,7 +2423,7 @@ def get_default_conv_template(self, model_path: str) -> Conversation:
24232423

24242424

24252425
class DBRXAdapter(BaseModelAdapter):
2426-
"""The model adapter for Cohere"""
2426+
"""The model adapter for Databricks"""
24272427

24282428
def match(self, model_path: str):
24292429
return model_path in ["dbrx-instruct"]

fastchat/model/model_registry.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -195,17 +195,17 @@ def get_model_info(name: str) -> ModelInfo:
195195
)
196196

197197
register_model_info(
198-
["command-r-plus"],
199-
"Command-R-Plus",
198+
["command-r-plus", "command-r-plus-04-2024"],
199+
"Command R+",
200200
"https://txt.cohere.com/command-r-plus-microsoft-azure/",
201-
"Command-R Plus by Cohere",
201+
"Command R+ by Cohere",
202202
)
203203

204204
register_model_info(
205-
["command-r"],
206-
"Command-R",
205+
["command-r", "command-r-03-2024", "command-r-08-2024"],
206+
"Command R",
207207
"https://txt.cohere.com/command-r/",
208-
"Command-R by Cohere",
208+
"Command R by Cohere",
209209
)
210210

211211
register_model_info(

0 commit comments

Comments
 (0)