Skip to content

Commit e980e71

Browse files
committed
fix(mlx.ts): add repetition_penalty option to generateCommitMessage method for improved model behavior
1 parent 6cb3f5c commit e980e71

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

out/cli.cjs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44542,6 +44542,7 @@ var MLXEngine = class {
4454244542
messages,
4454344543
temperature: 0,
4454444544
top_p: 0.1,
44545+
repetition_penalty: 1.5,
4454544546
stream: false
4454644547
};
4454744548
try {

out/github-action.cjs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63341,6 +63341,7 @@ var MLXEngine = class {
6334163341
messages,
6334263342
temperature: 0,
6334363343
top_p: 0.1,
63344+
repetition_penalty: 1.5,
6334463345
stream: false
6334563346
};
6334663347
try {

src/engine/mlx.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,14 +18,15 @@ export class MLXEngine implements AiEngine {
1818
headers: { 'Content-Type': 'application/json' }
1919
});
2020
}
21-
21+
2222
async generateCommitMessage(
2323
messages: Array<OpenAI.Chat.Completions.ChatCompletionMessageParam>):
2424
Promise<string | undefined> {
2525
const params = {
2626
messages,
2727
temperature: 0,
2828
top_p: 0.1,
29+
repetition_penalty: 1.5,
2930
stream: false
3031
};
3132
try {

0 commit comments

Comments
 (0)