diff --git a/__init__.py b/__init__.py index 901f607..26bbd53 100644 --- a/__init__.py +++ b/__init__.py @@ -24,10 +24,10 @@ WEB_DIRECTORY = "./web" -from .nodes import LoadAudio,PreViewAudio,LoadSRT,FishSpeech_INFER,FishSpeech_INFER_SRT +from .nodes import LoadAudioPath,PreViewAudio,LoadSRT,FishSpeech_INFER,FishSpeech_INFER_SRT NODE_CLASS_MAPPINGS = { - "LoadAudio": LoadAudio, + "LoadAudioPath": LoadAudioPath, "PreViewAudio": PreViewAudio, "LoadSRT": LoadSRT, "FishSpeech_INFER": FishSpeech_INFER, @@ -36,7 +36,7 @@ # A dictionary that contains the friendly/humanly readable titles for the nodes NODE_DISPLAY_NAME_MAPPINGS = { - "LoadAudio": "AudioLoader", + "LoadAudioPath": "LoadAudioPath", "PreViewAudio": "PreView Audio", "LoadSRT": "SRT FILE Loader", "FishSpeech_INFER": "FishSpeech Inference", diff --git a/nodes.py b/nodes.py index 9b5e225..a191e98 100644 --- a/nodes.py +++ b/nodes.py @@ -23,7 +23,7 @@ class FishSpeech_INFER_SRT: def INPUT_TYPES(s): return {"required":{ "text":("SRT",), - "prompt_audio": ("AUDIO",), + "prompt_audio": ("AUDIOPATH",), "prompt_text":("SRT",), "if_mutiple_speaker":("BOOLEAN",{ "default": False @@ -71,7 +71,7 @@ def INPUT_TYPES(s): }} CATEGORY = "AIFSH_FishSpeech" - RETURN_TYPES = ('AUDIO',) + RETURN_TYPES = ('AUDIOPATH',) OUTPUT_NODE = False FUNCTION = "get_tts_wav" @@ -216,7 +216,7 @@ class FishSpeech_INFER: @classmethod def INPUT_TYPES(s): return {"required":{ - "prompt_audio": ("AUDIO",), + "prompt_audio": ("AUDIOPATH",), "text":("STRING",{ "multiline": True, "default": "你好啊,世界!" @@ -268,7 +268,7 @@ def INPUT_TYPES(s): }} CATEGORY = "AIFSH_FishSpeech" - RETURN_TYPES = ('AUDIO',) + RETURN_TYPES = ('AUDIOPATH',) OUTPUT_NODE = False FUNCTION = "get_tts_wav" @@ -329,7 +329,7 @@ class PreViewAudio: @classmethod def INPUT_TYPES(s): return {"required": - {"audio": ("AUDIO",),} + {"audio": ("AUDIOPATH",),} } CATEGORY = "AIFSH_FishSpeech" @@ -347,7 +347,7 @@ def load_audio(self, audio): audio_root = os.path.basename(tmp_path) return {"ui": {"audio":[audio_name,audio_root]}} -class LoadAudio: +class LoadAudioPath: @classmethod def INPUT_TYPES(s): files = [f for f in os.listdir(input_path) if os.path.isfile(os.path.join(input_path, f)) and f.split('.')[-1] in ["wav", "mp3","WAV","flac","m4a"]] @@ -357,7 +357,7 @@ def INPUT_TYPES(s): CATEGORY = "AIFSH_FishSpeech" - RETURN_TYPES = ("AUDIO",) + RETURN_TYPES = ("AUDIOPATH",) FUNCTION = "load_audio" def load_audio(self, audio):