Add NAI Diffusion V4

This commit is contained in:
Cohee
2024-12-22 15:39:41 +02:00
parent 713443d234
commit e6107ad447
3 changed files with 32 additions and 3 deletions

View File

@@ -1985,6 +1985,10 @@ async function loadNovelModels() {
}
return [
{
value: 'nai-diffusion-4-curated-preview',
text: 'NAI Diffusion Anime V4 (Curated Preview)',
},
{
value: 'nai-diffusion-3',
text: 'NAI Diffusion Anime V3',
@@ -2049,7 +2053,7 @@ async function loadSchedulers() {
schedulers = await getAutoRemoteSchedulers();
break;
case sources.novel:
schedulers = ['N/A'];
schedulers = ['karras', 'native', 'exponential', 'polyexponential'];
break;
case sources.vlad:
schedulers = ['N/A'];
@@ -3158,6 +3162,7 @@ async function generateNovelImage(prompt, negativePrompt, signal) {
prompt: prompt,
model: extension_settings.sd.model,
sampler: extension_settings.sd.sampler,
scheduler: extension_settings.sd.scheduler,
steps: steps,
scale: extension_settings.sd.scale,
width: width,

View File

@@ -274,7 +274,7 @@
<select id="sd_sampler"></select>
</div>
<div class="flex1" data-sd-source="comfy,auto">
<div class="flex1" data-sd-source="comfy,auto,novel">
<label for="sd_scheduler" data-i18n="Scheduler">Scheduler</label>
<select id="sd_scheduler"></select>
</div>