diff --git a/nodes.py b/nodes.py index 0d9a96b..5bf16ea 100644 --- a/nodes.py +++ b/nodes.py @@ -43,7 +43,7 @@ class SVDModelLoader: FUNCTION = "load_svd_model" - CATEGORY = "Comfy Stable Video Diffusion" + CATEGORY = "ComfyUI Stable Video Diffusion" def load_svd_model(self, checkpoint, num_frames, num_steps, device): if self.svd_model is not None: @@ -90,13 +90,10 @@ class SVDSampler: } RETURN_TYPES = ("LATENT",) - #RETURN_NAMES = ("image_output_name",) FUNCTION = "sample_video" - #OUTPUT_NODE = False - - CATEGORY = "Comfy Stable Video Diffusion" + CATEGORY = "ComfyUI Stable Video Diffusion" def sample_video(self, image, model, motion_bucket_id, fps_id, cond_aug, seed, device): # convert image torch tensor to PIL image @@ -218,7 +215,7 @@ class SVDDecoder: FUNCTION = "decode" - CATEGORY = "Comfy Stable Video Diffusion" + CATEGORY = "ComfyUI Stable Video Diffusion" def decode(self, samples_z, model, decoding_t, device): with torch.no_grad(): @@ -243,7 +240,7 @@ NODE_CLASS_MAPPINGS = { # A dictionary that contains the friendly/humanly readable titles for the nodes NODE_DISPLAY_NAME_MAPPINGS = { - "SVDModelLoader" : "Load SVD Model", - "SVDSampler": "SVD Sampler", - "SVDDecoder": "SVD Decoder", + "SVDModelLoader" : "Load Stable Video Diffusion Model", + "SVDSampler": "Stable Video Diffusion Sampler", + "SVDDecoder": "Stable Video Diffusion Decoder", } \ No newline at end of file