From e03a168bdae7ee32f9b62202e08ff69dd50c718e Mon Sep 17 00:00:00 2001
From: BobRandomNumber <lordhigh@gmail.com>
Date: Mon, 28 Apr 2025 12:12:30 -0400
Subject: [PATCH] change DiaTest to DiaTTS

---
 nodes.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index dbfa578..f2dccd2 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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",
-}
\ No newline at end of file
+}