Merge pull request #3881 from DreamGenX/dg_lucid

DreamGen Lucid
This commit is contained in:
Cohee
2025-04-19 22:01:37 +03:00
committed by GitHub
4 changed files with 6 additions and 23 deletions

View File

@@ -928,6 +928,10 @@ export function getCurrentDreamGenModelTokenizer() {
return tokenizers.YI;
} else if (model.id.startsWith('opus-v1-xl')) {
return tokenizers.LLAMA;
} else if (model.id.startsWith('lucid-v1-medium')) {
return tokenizers.NEMO;
} else if (model.id.startsWith('lucid-v1-extra-large')) {
return tokenizers.LLAMA3;
} else {
return tokenizers.MISTRAL;
}