26 lines
1.3 KiB
Python
26 lines
1.3 KiB
Python
from .nodes.apply_scaleu_model_node import ApplyScaleUModelNode
|
|
from .nodes.load_instance_scaleu_node import LoadInstanceScaleUNode
|
|
from .nodes.load_instance_fusers_node import LoadInstanceFusersNode
|
|
from .nodes.load_instance_positionnet_node import LoadInstancePositionNetNode
|
|
from .nodes.instance_diffusion_tracking_prompt_node import InstanceDiffusionTrackingPromptNode
|
|
from .nodes.download_and_load_models import DownloadInstanceDiffusionModels
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ApplyScaleUModelNode": ApplyScaleUModelNode,
|
|
"LoadInstanceScaleUNode": LoadInstanceScaleUNode,
|
|
"LoadInstancePositionNetModel": LoadInstancePositionNetNode,
|
|
"LoadInstanceFusersNode": LoadInstanceFusersNode,
|
|
"InstanceDiffusionTrackingPrompt": InstanceDiffusionTrackingPromptNode,
|
|
"DownloadInstanceDiffusionModels": DownloadInstanceDiffusionModels
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ApplyScaleUModelNode": "Apply Instance Diffusion ScaleU",
|
|
"LoadInstancePositionNetModel": "Load Instance PositionNet Model",
|
|
"LoadInstanceScaleUModel": "Load Instance ScaleU Model",
|
|
"LoadInstanceFusersNode": "Load Instance Fusers Model",
|
|
"InstanceDiffusionTrackingPrompt": "Instance Diffusion Tracking Prompt",
|
|
"DownloadInstanceDiffusionModels": "(Down)Load Instance Diffusion Models"
|
|
}
|