NovelAI: Increase tier token limits

027d7107160f280f74ed580d57eb6940ee554fd8

Cohee <18619528+Cohee1207@users.noreply.github.com>

1 files changed, +7 -7Showing whitespace changes
public/scripts/nai-settings.js+7 -7
@@ -83,9 +83,9 @@ export function setNovelData(data) {
83export function getKayraMaxContextTokens() {83export function getKayraMaxContextTokens() {
84 switch (novel_data?.tier) {84 switch (novel_data?.tier) {
85 case 1:85 case 1:
86 return 3072;86 return 4096;
87 case 2:87 case 2:
88 return 6144;88 return 8192;
89 case 3:89 case 3:
90 return 8192;90 return 8192;
91 }91 }
@@ -93,14 +93,14 @@ export function getKayraMaxContextTokens() {
93 return null;93 return null;
94}94}
9595
96export function getKayraMaxResponseTokens() {96export function getNovelMaxResponseTokens() {
97 switch (novel_data?.tier) {97 switch (novel_data?.tier) {
98 case 1:98 case 1:
99 return 100;99 return 150;
100 case 2:100 case 2:
101 return 100;
102 case 3:
103 return 150;101 return 150;
102 case 3:
103 return 250;
104 }104 }
105105
106 return maximum_output_length;106 return maximum_output_length;
@@ -546,7 +546,7 @@ export function getNovelGenerationData(finalPrompt, settings, maxLength, isImper
546 finalPrompt = '<|startoftext|><|reserved_special_token81|>' + finalPrompt;546 finalPrompt = '<|startoftext|><|reserved_special_token81|>' + finalPrompt;
547 }547 }
548548
549 const adjustedMaxLength = (isKayra || isErato) ? getKayraMaxResponseTokens() : maximum_output_length;549 const adjustedMaxLength = (isKayra || isErato) ? getNovelMaxResponseTokens() : maximum_output_length;
550550
551 return {551 return {
552 'input': finalPrompt,552 'input': finalPrompt,