22 lines
819 B
Python
22 lines
819 B
Python
from .use_everywhere import UseEverywhere
|
|
|
|
types = {
|
|
"MODEL" : ("MODEL",),
|
|
"VAE" : ("VAE",),
|
|
"CLIP" : ("CLIP",),
|
|
"LATENT" : ("LATENT", ),
|
|
"IMAGE" : ("IMAGE",),
|
|
"CONDITIONING" : ("CONDITIONING",),
|
|
"CHECKPOINT" : ("MODEL", "CLIP", "VAE"),
|
|
}
|
|
|
|
NODE_CLASS_MAPPINGS = { f"UE {t}" : type(f"UE {t}", (UseEverywhere,), { "RETURN_TYPES":types[t] }) for t in types }
|
|
__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) |