43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
try:
|
|
from custom_nodes.cg_custom_core import CC_VERSION
|
|
except:
|
|
print("cg_custom_core not found - will try to install - you may need to restart afterwards")
|
|
from .install import installer
|
|
import os
|
|
import folder_paths
|
|
application_root_directory = os.path.dirname(folder_paths.__file__)
|
|
installer(os.path.join(application_root_directory,"custom_nodes"))
|
|
|
|
from .use_everywhere import UseEverywhere, UseSomewhere, SeedEverywhere
|
|
|
|
types = {
|
|
"MODEL" : ("MODEL",),
|
|
"VAE" : ("VAE",),
|
|
"CLIP" : ("CLIP",),
|
|
"LATENT" : ("LATENT", ),
|
|
"IMAGE" : ("IMAGE",),
|
|
"MASK" : ("MASK",),
|
|
"CONDITIONING" : ("CONDITIONING",),
|
|
"INT" : ("INT",),
|
|
"CHECKPOINT" : ("MODEL", "CLIP", "VAE"),
|
|
}
|
|
|
|
NODE_CLASS_MAPPINGS = { "Seed Everywhere": SeedEverywhere }
|
|
|
|
for t in types:
|
|
NODE_CLASS_MAPPINGS[f"UE {t}"] = type(f"UE {t}", (UseEverywhere,), { "RETURN_TYPES":types[t] })
|
|
NODE_CLASS_MAPPINGS[f"UE? {t}"] = type(f"UE {t}", (UseSomewhere,), { "RETURN_TYPES":types[t] })
|
|
|
|
from .use_everywhere import AnythingEverywhere, AnythingSomewhere
|
|
NODE_CLASS_MAPPINGS["Anything Everywhere"] = AnythingEverywhere
|
|
NODE_CLASS_MAPPINGS["Anything Everywhere?"] = AnythingSomewhere
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS']
|
|
|
|
import os, shutil
|
|
import folder_paths
|
|
module_js_directory = os.path.join(os.path.dirname(os.path.realpath(__file__)), "js")
|
|
application_root_directory = os.path.dirname(folder_paths.__file__)
|
|
application_web_extensions_directory = os.path.join(application_root_directory, "web", "extensions", "cg-nodes")
|
|
|
|
shutil.copytree(module_js_directory, application_web_extensions_directory, dirs_exist_ok=True) |