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
commit 590ad6235b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2010,6 +2010,10 @@ async function loadVladModels() {
async function loadNovelModels() {
return [
{
value: 'nai-diffusion-4-full',
text: 'NAI Diffusion Anime V4 (Full)',
},
{
value: 'nai-diffusion-4-curated-preview',
text: 'NAI Diffusion Anime V4 (Curated Preview)',
@ -2333,7 +2337,10 @@ function processReply(str) {
str = str.replaceAll('“', '');
str = str.replaceAll('\n', ', ');
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.trim();
@ -3227,7 +3234,8 @@ function getNovelParams() {
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_dyn = false;
}