Implement suggestions

This commit is contained in:
DreamGenX
2024-03-07 17:28:38 +01:00
parent 5c410986a4
commit bc8d41b530
3 changed files with 4 additions and 11 deletions

View File

@@ -376,8 +376,8 @@ export function getCurrentOpenRouterModelTokenizer() {
}
export function getCurrentDreamGenModelTokenizer() {
const modelId = textgen_settings.openrouter_model;
const model = openRouterModels.find(x => x.id === modelId);
const modelId = textgen_settings.dreamgen_model;
const model = dreamGenModels.find(x => x.id === modelId);
if (model.id.startsWith('opus-v1-sm')) {
return tokenizers.MISTRAL;
} else if (model.id.startsWith('opus-v1-lg')) {