Merge branch 'coqui_cuda'

jersou · Oct 15, 2024 · d61b689 · d61b689
2 parents 94c9ebc + 750be02
commit d61b689
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -253,6 +253,7 @@ Options:
  -g, --open-ai-model                OpenAi model : tts-1, tts-1-hd                                [default: "tts-1"]
  -p, --open-ai-voice                OpenAi voice : alloy, echo, fable, onyx, nova, shimmer         [default: "onyx"]
      --use-coqui-tts                use coqui TTS                                                   [default: false]
+     --coqui-tts-use-cuda           enable CUDA in coqui TTS                                        [default: false]
      --coqui-tts-model              coqui TTS model       [default: "tts_models/multilingual/multi-dataset/xtts_v2"]
      --coqui-tts-language-idx       coqui TTS language_idx                                           [default: "fr"]
      --coqui-tts-speaker-idx        coqui TTS speaker_idx                                  [default: "Abrahan Mack"]
@@ -412,6 +413,7 @@ File format (all the properties are optionals) :
   "rssEpisodeNumbers": false,
   "useCoquiTts": false,
   "coquiTtsModel": "tts_models/multilingual/multi-dataset/xtts_v2",
+  "coquiUseCuda": false,
   "coquiTtsLanguageIdx": "fr",
   "coquiTtsSpeakerIdx": "Abrahan Mack",
   "port": 5555,

diff --git a/generate/coqui_tts.ts b/generate/coqui_tts.ts
@@ -33,6 +33,9 @@ export async function generate_audio_with_coqui(
     if (opt.coquiTtsLanguageIdx) {
       cmd.push("--language_idx", opt.coquiTtsLanguageIdx);
     }
+    if (opt.coquiTtsUseCuda) {
+      cmd.push("--use_cuda", "1");
+    }
     if (opt.coquiTtsSpeakerIdx) {
       cmd.push("--speaker_idx", opt.coquiTtsSpeakerIdx);
     }

diff --git a/studio_pack_generator.ts b/studio_pack_generator.ts
@@ -141,6 +141,9 @@ export class StudioPackGenerator {
   @help("use coqui TTS")
   useCoquiTts = false;
 
+  @help("enable CUDA in coqui TTS")
+  coquiTtsUseCuda = false;
+
   @help("coqui TTS model")
   coquiTtsModel = "tts_models/multilingual/multi-dataset/xtts_v2";