Skip to content

Commit 53c591f

Browse files
authored
Adds support for StarChat for autocomplete (#847)
Updates the handling of template generation to treat StarChat models the same as StarCoder models.
1 parent 808af46 commit 53c591f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

core/autocomplete/templates.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ export function getTemplateForModel(model: string): AutocompleteTemplate {
4949
if (
5050
lowerCaseModel.includes("starcoder") ||
5151
lowerCaseModel.includes("star-coder") ||
52+
lowerCaseModel.includes("starchat") ||
5253
lowerCaseModel.includes("stable")
5354
) {
5455
return stableCodeFimTemplate;

0 commit comments

Comments
 (0)