Registered nodes with ComfyUI

This commit is contained in:
Brody Childs
2024-02-23 16:09:26 -05:00
parent 129e594991
commit 8299bce2ca
2 changed files with 47 additions and 5 deletions
+14 -5
View File
@@ -1,25 +1,34 @@
# MBM's Music Visualizer
# Imports
from .mbmMusicVisualizer import MusicVisualizer
from .mbmAudioFeatureCalculator import AudioFeatureCalculator
from .mbmAudioLoader import AudioLoader
from .mbmMusicVisualizer import MusicVisualizer
from .mbmPromptSequence import PromptSequenceBuilder
from .mbmPromptSequenceInterpolator import PromptSequenceInterpolator
from .mbmImageConcat import ImageConcatenator
# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
# "id": Class
"mbmMusicVisuzalizer": MusicVisualizer,
"mbmAudioFeatureCalculator": AudioFeatureCalculator,
"mbmAudioLoader": AudioLoader,
"mbmPromptSequenceBuilder": PromptSequenceBuilder
"mbmMusicVisuzalizer": MusicVisualizer,
"mbmPromptSequenceBuilder": PromptSequenceBuilder,
"mbmPromptSequenceInterpolator": PromptSequenceInterpolator,
"mbmImageConcat": ImageConcatenator
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
# "id": "readable"
"mbmMusicVisuzalizer": "Music Visualizer",
"mbmAudioFeatureCalculator": "Audio Feature Calculator",
"mbmAudioLoader": "Audio Loader",
"mbmPromptSequenceBuilder": "Prompt Sequence Builder"
"mbmMusicVisuzalizer": "Music Visualizer",
"mbmPromptSequenceBuilder": "Prompt Sequence Builder",
"mbmPromptSequenceInterpolator": "Prompt Sequence Interpolator",
"mbmImageConcat": "Image Concatenator"
}
# Export
+33
View File
@@ -0,0 +1,33 @@
# MBM's Music Visualizer: Image Concatenator
# Concatenates ComfyUI image set tensors into a single set tensor.
# Imports
import torch
# Classes
class ImageConcatenator:
"""
Concatenates ComfyUI image set tensors into a single set tensor.
"""
# Class Constants
RETURN_TYPES = ("IMAGE", )
RETURN_NAMES = ("IMAGES", )
FUNCTION = "process"
CATEGORY = "MBMnodes/Images"
# Constructor
def __init__(self):
pass
# ComfyUI Functions
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"set_1": ("IMAGE", ),
"set_2": ("IMAGE", )
}
}
def process(self, set_1: torch.Tensor, set_2: torch.Tensor) -> torch.Tensor:
return (torch.cat((set_1, set_2), dim=0), )