File tree Expand file tree Collapse file tree 1 file changed +19
-0
lines changed
core/autocomplete/templating Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Original file line number Diff line number Diff line change @@ -71,6 +71,21 @@ const qwenCoderFimTemplate: AutocompleteTemplate = {
71
71
} ,
72
72
} ;
73
73
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
+
74
89
const codestralFimTemplate : AutocompleteTemplate = {
75
90
template : "[SUFFIX]{{{suffix}}}[PREFIX]{{{prefix}}}" ,
76
91
completionOptions : {
@@ -426,6 +441,10 @@ export function getTemplateForModel(model: string): AutocompleteTemplate {
426
441
return qwenCoderFimTemplate ;
427
442
}
428
443
444
+ if ( lowerCaseModel . includes ( "seed" ) && lowerCaseModel . includes ( "coder" ) ) {
445
+ return seedCoderFimTemplate ;
446
+ }
447
+
429
448
if (
430
449
lowerCaseModel . includes ( "starcoder" ) ||
431
450
lowerCaseModel . includes ( "star-coder" ) ||
You can’t perform that action at this time.
0 commit comments