79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
class RunwareAudioModelSearch:
|
|
"""Audio Model Search node for searching audio models"""
|
|
|
|
# Define the audio models by provider
|
|
AUDIO_MODELS = {
|
|
"ElevenLabs": [
|
|
"elevenlabs:1@1 (ElevenLabs Multilingual v2)",
|
|
"elevenlabs:2@1 (ElevenLabs Multilingual v2 Turbo)",
|
|
"elevenlabs:3@1 (ElevenLabs Monolingual v1)",
|
|
],
|
|
"KlingAI": [
|
|
"klingai:8@1 (KlingAI Audio)",
|
|
],
|
|
}
|
|
|
|
MODEL_PROVIDERS = [
|
|
"All",
|
|
"ElevenLabs",
|
|
"KlingAI",
|
|
]
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
allModels = cls._getAllModels()
|
|
defaultModel = "elevenlabs:1@1 (ElevenLabs Multilingual v2)"
|
|
|
|
return {
|
|
"required": {
|
|
"Model Search": ("STRING", {
|
|
"tooltip": "Search For Specific Audio Model By Name Or Code (eg: elevenlabs, klingai).",
|
|
}),
|
|
"Model Provider": (cls.MODEL_PROVIDERS, {
|
|
"tooltip": "Choose Audio Model Provider To Filter Results.",
|
|
"default": "ElevenLabs",
|
|
}),
|
|
"AudioList": (allModels, {
|
|
"tooltip": "Audio Model Results Will Show UP Here So You Could Choose From.",
|
|
"default": defaultModel,
|
|
}),
|
|
"Use Search Value": ("BOOLEAN", {
|
|
"tooltip": "When Enabled, the value you've set in the search input will be used instead.\n\nThis is useful in case the model search API is down or you prefer to set the model manually.",
|
|
"default": False,
|
|
"label_on": "Enabled",
|
|
"label_off": "Disabled",
|
|
}),
|
|
}
|
|
}
|
|
|
|
@classmethod
|
|
def _getAllModels(cls):
|
|
"""Get all models from all providers"""
|
|
allModels = []
|
|
for models in cls.AUDIO_MODELS.values():
|
|
allModels.extend(models)
|
|
return allModels
|
|
|
|
RETURN_TYPES = ("RUNWAREAUDIOMODEL",)
|
|
RETURN_NAMES = ("model",)
|
|
FUNCTION = "searchModels"
|
|
CATEGORY = "Runware"
|
|
DESCRIPTION = "Directly Search and Connect Audio Models to Runware Audio Inference Nodes In ComfyUI."
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, AudioList):
|
|
return True
|
|
|
|
def searchModels(self, **kwargs):
|
|
"""Search and return audio model"""
|
|
enableSearchValue = kwargs.get("Use Search Value", False)
|
|
searchInput = kwargs.get("Model Search")
|
|
|
|
if enableSearchValue:
|
|
modelAirCode = searchInput
|
|
else:
|
|
currentModel = kwargs.get("AudioList")
|
|
modelAirCode = currentModel.split(" (")[0]
|
|
|
|
return (modelAirCode,)
|