44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
from .sd15_unet_loader import SD15UNETLoader
|
|
from .sd15_vae_loader import SD15VAELoader
|
|
from .sd15_clip_loader import SD15CLIPLoader
|
|
from .sdxl_unet_loader import SDXLUNETLoader
|
|
from .sdxl_vae_loader import SDXLVAELoader
|
|
from .sdxl_clip_loader import SDXLCLIPLoader
|
|
from .sd15_combined_loader import CombinedDiffusersSD15Loader
|
|
from .sdxl_combined_loader import CombinedDiffusersSDXLLoader
|
|
from .combined_diffusers_loader import CombinedDiffusersLoader
|
|
from .diffusers_clip_loader import DiffusersClipLoader
|
|
from .diffusers_unet_loader import DiffusersUNETLoader
|
|
from .diffusers_vae_loader import DiffusersVAELoader
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SD15UNETLoader": SD15UNETLoader,
|
|
"SD15VAELoader": SD15VAELoader,
|
|
"SD15CLIPLoader": SD15CLIPLoader,
|
|
"SDXLUNETLoader": SDXLUNETLoader,
|
|
"SDXLVAELoader": SDXLVAELoader,
|
|
"SDXLCLIPLoader": SDXLCLIPLoader,
|
|
"CombinedDiffusersSD15Loader": CombinedDiffusersSD15Loader,
|
|
"CombinedDiffusersSDXLLoader": CombinedDiffusersSDXLLoader,
|
|
"CombinedDiffusersLoader": CombinedDiffusersLoader,
|
|
"DiffusersClipLoader": DiffusersClipLoader,
|
|
"DiffusersUNETLoader": DiffusersUNETLoader,
|
|
"DiffusersVAELoader": DiffusersVAELoader
|
|
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SD15UNETLoader": "SD1.5 UNET Loader",
|
|
"SD15VAELoader": "SD1.5 VAE Loader",
|
|
"SD15CLIPLoader": "SD1.5 CLIP Loader",
|
|
"SDXLUNETLoader": "SDXL UNET Loader",
|
|
"SDXLVAELoader": "SDXL VAE Loader",
|
|
"SDXLCLIPLoader": "SDXL CLIP Loader",
|
|
"CombinedDiffusersSD15Loader" : "Combined Diffusers SD15 Loader",
|
|
"CombinedDiffusersSDXLLoader" : "Combined Diffusers SDXL Loader",
|
|
"CombinedDiffusersLoader": "Combined Diffusers Loader",
|
|
"DiffusersClipLoader": "Diffusers CLIP Loader",
|
|
"DiffusersUNETLoader": "Diffusers UNET Loader",
|
|
"DiffusersVAELoader": "Diffusers VAE Loader"
|
|
}
|