From 05826d2e935f25a51e6ded78e0f7e014f6bb44a9 Mon Sep 17 00:00:00 2001
From: chenyunda218 <chenyunda218@gmail.com>
Date: Mon, 29 Jul 2024 10:45:38 +0800
Subject: [PATCH] feat: cosyvoicetts

---
 src/blackbox/blackbox_factory.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py
index 1c9e049..8565eb1 100644
--- a/src/blackbox/blackbox_factory.py
+++ b/src/blackbox/blackbox_factory.py
@@ -97,6 +97,11 @@ def chat_llama_loader():
     from .chat_llama import ChatLLaMA
     return Injector().get(ChatLLaMA)
 
+@model_loader(lazy=blackboxConf.lazyloading)
+def cosyvoicetts_loader():
+    from .cosyvoicetts import CosyVoiceTTS
+    return Injector().get(CosyVoiceTTS)
+
 @singleton
 class BlackboxFactory:
     models = {}
@@ -118,6 +123,7 @@ class BlackboxFactory:
         self.models["vlms"] = vlms_loader
         self.models["chat"] = chat_loader
         self.models["chat_llama"] = chat_llama_loader
+        self.models["cosyvoicetts"] = cosyvoicetts_loader
 
     def __call__(self, *args, **kwargs):
         return self.processing(*args, **kwargs)