Commit06eb07f

committed

Added space for better infills and proper stop token

1 parent9288c45 commit06eb07fCopy full SHA for 06eb07f

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@ export function adaptPrompt(args: { format: ModelFormat, prefix: string, suffix:`
`19`	`19`	`}`
`20`	`20`
`21`	`21`	`// Stable code FIM`
`22`		`-if(args.format==='stable-code'){`
	`22`	`+if(args.format==='stable-code'){`
`23`	`23`	`return{`
`24`	`24`	prompt:`<fim_prefix>${args.prefix}<fim_suffix>${args.suffix}<fim_middle>`,
`25`	`25`	stop:[`<\|endoftext\|>`]
`@@ -28,7 +28,7 @@ export function adaptPrompt(args: { format: ModelFormat, prefix: string, suffix:`
`28`	`28`
`29`	`29`	`// Codellama FIM`
`30`	`30`	`return{`
`31`		-prompt:`<PRE>${args.prefix} <SUF>${args.suffix} <MID>`,
`32`		-stop:[`<PRE>`,`<SUF>`,`<MID>`,`<END>`]
	`31`	+prompt:`<PRE>${args.prefix} <SUF>${args.suffix} <MID>`,
	`32`	+stop:[`<END>`,`<EOD>`,`<EOT>`]
`33`	`33`	`};`
`34`	`34`	`}`

Comments

(0)