forked from asagi4/AIT
-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
23 lines (21 loc) · 1.16 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from .AITemplate.AITemplate import AITemplateLoader, AITemplateControlNetLoader, AITemplateVAEDecode, AITemplateVAEEncode, VAEEncodeForInpaint, AITemplateEmptyLatentImage, AITemplateLatentUpscale
NODE_CLASS_MAPPINGS = {
"AITemplateLoader": AITemplateLoader,
"AITemplateControlNetLoader": AITemplateControlNetLoader,
"AITemplateVAEDecode": AITemplateVAEDecode,
"AITemplateVAEEncode": AITemplateVAEEncode,
"AITemplateVAEEncodeForInpaint": VAEEncodeForInpaint,
"AITemplateEmptyLatentImage": AITemplateEmptyLatentImage,
"AITemplateLatentUpscale": AITemplateLatentUpscale,
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
"AITemplateLoader": "Load AITemplate",
"AITemplateControlNetLoader": "Load AITemplate (ControlNet)",
"AITemplateVAELoader": "Load AITemplate (VAE)",
"AITemplateVAEDecode": "VAE Decode (AITemplate)",
"AITemplateVAEEncode": "VAE Encode (AITemplate)",
"AITemplateVAEEncodeForInpaint": "VAE Encode (AITemplate, Inpaint)",
"AITemplateEmptyLatentImage": "Empty Latent Image (AITemplate)",
"AITemplateLatentUpscale": "Upscale Latent Image (AITemplate)",
}