Merge pull request #3586 from DAurielS/novelai-image-update

Added support for NovelAI Diffusion V4 Full via API to Image Generation extension.
This commit is contained in:
Cohee
2025-03-02 01:05:35 +02:00
committed by GitHub

View File

@ -2010,6 +2010,10 @@ async function loadVladModels() {
async function loadNovelModels() { async function loadNovelModels() {
return [ return [
{
value: 'nai-diffusion-4-full',
text: 'NAI Diffusion Anime V4 (Full)',
},
{ {
value: 'nai-diffusion-4-curated-preview', value: 'nai-diffusion-4-curated-preview',
text: 'NAI Diffusion Anime V4 (Curated Preview)', text: 'NAI Diffusion Anime V4 (Curated Preview)',
@ -2333,7 +2337,10 @@ function processReply(str) {
str = str.replaceAll('“', ''); str = str.replaceAll('“', '');
str = str.replaceAll('\n', ', '); str = str.replaceAll('\n', ', ');
str = str.normalize('NFD'); str = str.normalize('NFD');
str = str.replace(/[^a-zA-Z0-9.,:_(){}<>[\]\-']+/g, ' ');
// Strip out non-alphanumeric characters barring model syntax exceptions
str = str.replace(/[^a-zA-Z0-9.,:_(){}<>[\]\-'|#]+/g, ' ');
str = str.replace(/\s+/g, ' '); // Collapse multiple whitespaces into one str = str.replace(/\s+/g, ' '); // Collapse multiple whitespaces into one
str = str.trim(); str = str.trim();
@ -3227,7 +3234,8 @@ function getNovelParams() {
extension_settings.sd.scheduler = 'karras'; extension_settings.sd.scheduler = 'karras';
} }
if (extension_settings.sd.sampler === 'ddim' || extension_settings.sd.model === 'nai-diffusion-4-curated-preview') { if (extension_settings.sd.sampler === 'ddim' ||
['nai-diffusion-4-curated-preview', 'nai-diffusion-4-full'].includes(extension_settings.sd.model)) {
sm = false; sm = false;
sm_dyn = false; sm_dyn = false;
} }