32 lines
1.2 KiB
Python
32 lines
1.2 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
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SD15UNETLoader": SD15UNETLoader,
|
|
"SD15VAELoader": SD15VAELoader,
|
|
"SD15CLIPLoader": SD15CLIPLoader,
|
|
"SDXLUNETLoader": SDXLUNETLoader,
|
|
"SDXLVAELoader": SDXLVAELoader,
|
|
"SDXLCLIPLoader": SDXLCLIPLoader,
|
|
"CombinedDiffusersSD15Loader": CombinedDiffusersSD15Loader,
|
|
"CombinedDiffusersSDXLLoader": CombinedDiffusersSDXLLoader,
|
|
|
|
}
|
|
|
|
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"
|
|
}
|