DreamGen Lucid

dc6407ee8fb5b1a0fad27e29d2b8cbf2901acf72

DreamGenX <x@dreamgen.com>

4 files changed, +6 -22Showing whitespace changes
public/index.html+2 -2
@@ -1416,7 +1416,7 @@
14161416 </div>
14171417 </div>
14181418
14191419 <div data-tg-type="aphrodite, ooba, koboldcpp, tabby, llamacpp, dreamgen" id="dryBlock" class="wide100p">
14201420 <h4 class="wide100p textAlignCenter" title="DRY penalizes tokens that would extend the end of the input into a sequence that has previously occurred in the input. Set multiplier to 0 to disable." data-i18n="[title]DRY_Repetition_Penalty_desc">
14211421 <label data-i18n="DRY Repetition Penalty">DRY Repetition Penalty</label>
14221422 <a href="https://github.com/oobabooga/text-generation-webui/pull/5677" target="_blank">
@@ -1571,7 +1571,7 @@
15711571 <div class="fa-solid fa-circle-info opacity50p " data-i18n="[title]Add the bos_token to the beginning of prompts. Disabling this can make the replies more creative" title="Add the bos_token to the beginning of prompts. Disabling this can make the replies more creative."></div>
15721572 </label>
15731573 </label>
15741574 <label data-tg-type="ooba, llamacpp, tabby, koboldcpp, dreamgen" class="checkbox_label flexGrow flexShrink" for="ban_eos_token_textgenerationwebui">
15751575 <input type="checkbox" id="ban_eos_token_textgenerationwebui" />
15761576 <label>
15771577 <small data-i18n="Ban EOS Token">Ban EOS Token</small>
public/scripts/textgen-models.js+4 -0
@@ -923,6 +923,10 @@ export function getCurrentDreamGenModelTokenizer() {
923923 return tokenizers.YI;
924924 } else if (model.id.startsWith('opus-v1-xl')) {
925925 return tokenizers.LLAMA;
926+ } else if (model.id.startsWith('lucid-v1-medium')) {
927+ return tokenizers.NEMO;
928+ } else if (model.id.startsWith('lucid-v1-extra-large')) {
929+ return tokenizers.LLAMA3;
926930 } else {
927931 return tokenizers.MISTRAL;
928932 }
src/constants.js+0 -17
@@ -268,23 +268,6 @@ export const FEATHERLESS_KEYS = [
268268 'guided_whitespace_pattern',
269269];
270270
271-// https://dreamgen.com/docs/api#openai-text
272-export const DREAMGEN_KEYS = [
273- 'model',
274- 'prompt',
275- 'max_tokens',
276- 'temperature',
277- 'top_p',
278- 'top_k',
279- 'min_p',
280- 'repetition_penalty',
281- 'frequency_penalty',
282- 'presence_penalty',
283- 'stop',
284- 'stream',
285- 'minimum_message_content_tokens',
286-];
287-
288271// https://docs.together.ai/reference/completions
289272export const TOGETHERAI_KEYS = [
290273 'model',
src/endpoints/backends/text-completions.js+0 -3
@@ -340,9 +340,6 @@ router.post('/generate', async function (request, response) {
340340 }
341341
342342 if (request.body.api_type === TEXTGEN_TYPES.DREAMGEN) {
343- request.body = _.pickBy(request.body, (_, key) => DREAMGEN_KEYS.includes(key));
344- // NOTE: DreamGen sometimes get confused by the unusual formatting in the character cards.
345- request.body.stop?.push('### User', '## User');
346343 args.body = JSON.stringify(request.body);
347344 }
348345