Skip to content

Commit efc3d1d

Browse files
Merge pull request #5608 from Orion-zhen/main
✨ add seed coder fim template
2 parents 03f1d28 + ea3bf04 commit efc3d1d

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

core/autocomplete/templating/AutocompleteTemplate.ts

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,21 @@ const qwenCoderFimTemplate: AutocompleteTemplate = {
7171
},
7272
};
7373

74+
const seedCoderFimTemplate: AutocompleteTemplate = {
75+
template: "<[fim-prefix]>{{{prefix}}}<[fim-suffix]>{{{suffix}}}<[fim-middle]>",
76+
completionOptions: {
77+
stop: [
78+
"<[end▁of▁sentence]>",
79+
"<[fim-prefix]>",
80+
"<[fim-middle]>",
81+
"<[fim-suffix]>",
82+
"<[PAD▁TOKEN]>",
83+
"<[SEP▁TOKEN]>",
84+
"<[begin▁of▁sentence]>",
85+
],
86+
},
87+
};
88+
7489
const codestralFimTemplate: AutocompleteTemplate = {
7590
template: "[SUFFIX]{{{suffix}}}[PREFIX]{{{prefix}}}",
7691
completionOptions: {
@@ -426,6 +441,10 @@ export function getTemplateForModel(model: string): AutocompleteTemplate {
426441
return qwenCoderFimTemplate;
427442
}
428443

444+
if (lowerCaseModel.includes("seed") && lowerCaseModel.includes("coder")) {
445+
return seedCoderFimTemplate;
446+
}
447+
429448
if (
430449
lowerCaseModel.includes("starcoder") ||
431450
lowerCaseModel.includes("star-coder") ||

0 commit comments

Comments
 (0)