Skip to content

Commit

Permalink
Improves the fallback fill-in-middle prompt template
Browse files Browse the repository at this point in the history
  • Loading branch information
AnnoyingTechnology committed Dec 13, 2024
1 parent 1b0b296 commit 620b0c8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion core/llm/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -439,7 +439,7 @@ export abstract class BaseLLM implements ILLM {
): AsyncGenerator<string> {
const { completionOptions, log } = this._parseCompletionOptions(options);

const madeUpFimPrompt = `${prefix}<FIM>${suffix}`;
const madeUpFimPrompt = `<|fim_prefix|>${prefix}<|fim_suffix|>${suffix}<|fim_middle|>`;
if (log) {
if (this.writeLog) {
await this.writeLog(
Expand Down

0 comments on commit 620b0c8

Please sign in to comment.