change DiaTest to DiaTTS

This commit is contained in:
BobRandomNumber
2025-04-28 12:12:30 -04:00
committed by GitHub
parent 8bc720d5be
commit e03a168bda
+3 -3
View File
@@ -60,7 +60,7 @@ class DiaLoader:
RETURN_TYPES = ("DIA_MODEL",)
RETURN_NAMES = ("dia_model",)
FUNCTION = "load_dia_model"
CATEGORY = "audio/DiaTest"
CATEGORY = "audio/DiaTTS"
def load_dia_model(self, ckpt_name: str):
"""Loads the safetensors weights, combines with embedded config, loads DAC, and prepares the Dia object."""
@@ -163,7 +163,7 @@ class DiaGenerate:
RETURN_TYPES = ("AUDIO",)
RETURN_NAMES = ("audio",)
FUNCTION = "generate_audio"
CATEGORY = "audio/DiaTest"
CATEGORY = "audio/DiaTTS"
def generate_audio(self, dia_model: Dia, text: str, max_tokens: int, cfg_scale: float, temperature: float, top_p: float, cfg_filter_top_k: int, speed_factor: float, seed: int):
"""Performs TTS generation using the provided Dia model object."""
@@ -248,4 +248,4 @@ NODE_CLASS_MAPPINGS = {
NODE_DISPLAY_NAME_MAPPINGS = {
"DiaLoader": "Dia 1.6b Loader",
"DiaGenerate": "Dia TTS Generate",
}
}