separated out BatchedTranscriptionEncoderNode

This commit is contained in:
SayanoAI
2024-06-27 11:44:03 -04:00
parent b17200147c
commit 953cc8e412
5 changed files with 174 additions and 143 deletions
+5 -3
View File
@@ -1,4 +1,4 @@
from .custom_nodes.stt import AudioTranscriptionNode
from .custom_nodes.stt import AudioTranscriptionNode, BatchedTranscriptionEncoderNode
from .custom_nodes.uvr import UVR5Node
from .custom_nodes.rvc import RVCNode
from .custom_nodes.loaders import DownloadAudio, LoadAudio, LoadWhisperModelNode, LoadRVCModelNode, LoadHubertModel, LoadPitchExtractionParams
@@ -26,7 +26,8 @@ NODE_CLASS_MAPPINGS = {
"MergeLatentBatches": MergeLatentBatches,
"ImageRepeatInterleavedNode": ImageRepeatInterleavedNode,
"LatentRepeatInterleavedNode": LatentRepeatInterleavedNode,
"DownloadAudio": DownloadAudio
"DownloadAudio": DownloadAudio,
"BatchedTranscriptionEncoderNode": BatchedTranscriptionEncoderNode
}
# A dictionary that contains the friendly/humanly readable titles for the nodes
@@ -46,5 +47,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"MergeImageBatches": "🌺Merge Image Batches",
"MergeLatentBatches": "🌺Merge Latent Batches",
"ImageRepeatInterleavedNode": "🌺Image Repeat Interleaved",
"LatentRepeatInterleavedNode": "🌺Latent Repeat Interleaved"
"LatentRepeatInterleavedNode": "🌺Latent Repeat Interleaved",
"BatchedTranscriptionEncoderNode": "🌺Batched CLIP Transcription Encode (Prompt)"
}