Skip to content

Commit

Permalink
πŸ› don't override max tokens for local models
Browse files Browse the repository at this point in the history
  • Loading branch information
sestinj committed Jun 6, 2024
1 parent a0d2820 commit feca6fa
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion core/autocomplete/completionProvider.ts
Original file line number Diff line number Diff line change
Expand Up @@ -637,7 +637,10 @@ export class CompletionProvider {
"llamafile",
"text-gen-webui",
];
if (LOCAL_PROVIDERS.includes(llm.providerName)) {
if (
!config.tabAutocompleteOptions?.maxPromptTokens &&
LOCAL_PROVIDERS.includes(llm.providerName)
) {
options.maxPromptTokens = 500;
}

Expand Down

0 comments on commit feca6fa

Please sign in to comment.