alihan/SwarmUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

also make cascade stage A as a VAE explicitly

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2024-09-19 00:05:10 +09:00
parent 9fe2849137
commit 1c44ab74c5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/Text2Image/T2IModelClassSorter.cs
View File

@@ -154,7 +154,7 @@ public class T2IModelClassSorter
return isSVD(h);
}});
// ====================== Stable Cascade ======================
Register(new() { ID = "stable-cascade-v1-stage-a", CompatClass = "stable-cascade-v1", Name = "Stable Cascade v1 (Stage A)", StandardWidth = 1024, StandardHeight = 1024, IsThisModelOfClass = (m, h) =>
Register(new() { ID = "stable-cascade-v1-stage-a/vae", CompatClass = "stable-cascade-v1", Name = "Stable Cascade v1 (Stage A)", StandardWidth = 1024, StandardHeight = 1024, IsThisModelOfClass = (m, h) =>
{
return isCascadeA(h) && !isCascadeB(h) && !isCascadeC(h);
}});
@@ -210,6 +210,7 @@ public class T2IModelClassSorter
Remaps["Flux.1-schnell/lora"] = "Flux.1-dev/lora";
Remaps["Flux.1-schnell/controlnet"] = "Flux.1-dev/controlnet";
Remaps["Flux.1-AE"] = "flux.1/vae";
Remaps["stable-cascade-v1-stage-a"] = "stable-cascade-v1-stage-a/vae";
// ====================== Random Other Models ======================
Register(new() { ID = "alt_diffusion_v1_512_placeholder", CompatClass = "alt_diffusion_v1", Name = "Alt-Diffusion", StandardWidth = 512, StandardHeight = 512, IsThisModelOfClass = (m, h) =>
{