diff --git a/spect_nodes.py b/spect_nodes.py index f1bd5c4..de5fbc0 100644 --- a/spect_nodes.py +++ b/spect_nodes.py @@ -11,7 +11,7 @@ class Spectrogram: "win_length": ("INT", {"default": -1, "min": -1, "max": 2**32}), "hop_length": ("INT", {"default": -1, "min": -1, "max": 2**32})}} - CATEGORY = "AudioProcessing/features" + CATEGORY = "AudioProcessing/spectrogram" RETURN_TYPES = ("SPECT", ) FUNCTION = "spectrogram" @@ -40,7 +40,7 @@ class InverseSpectrogram: def INPUT_TYPES(s): return {"required": {"spectrogram": ("SPECT",)}} - CATEGORY = "AudioProcessing/features" + CATEGORY = "AudioProcessing/spectrogram" RETURN_TYPES = ("AUDIO", ) FUNCTION = "inverse" @@ -63,7 +63,7 @@ class PowerSpectrogram: "win_length": ("INT", {"default": -1, "min": -1, "max": 2**32}), "hop_length": ("INT", {"default": -1, "min": -1, "max": 2**32})}} - CATEGORY = "AudioProcessing/features" + CATEGORY = "AudioProcessing/spectrogram" RETURN_TYPES = ("POWER_SPECT", ) FUNCTION = "spectrogram" @@ -94,7 +94,7 @@ class GriffinLim: "optional": {"n_iter": ("INT", {"default": 32, "min": 1, "max": 2**32}), "momentum": ("FLOAT", {"default": 0.99, "min": 0.0, "max": 1.0, "step": 0.01})}} - CATEGORY = "AudioProcessing/features" + CATEGORY = "AudioProcessing/spectrogram" RETURN_TYPES = ("AUDIO", ) FUNCTION = "griffin_lim" @@ -121,7 +121,7 @@ class MelSpectrogram: "win_length": ("INT", {"default": -1, "min": -1, "max": 2**32}), "hop_length": ("INT", {"default": -1, "min": -1, "max": 2**32})}} - CATEGORY = "AudioProcessing/features" + CATEGORY = "AudioProcessing/spectrogram" RETURN_TYPES = ("MEL_SPECT", ) FUNCTION = "spectrogram"