From 611ff1e8c02d584c97b053ec793696ceb9f5ed3f Mon Sep 17 00:00:00 2001
From: Jacky He <hekinghung@gmail.com>
Date: Tue, 2 Sep 2025 17:48:44 +0800
Subject: [PATCH] feat: make GPU selectable in get_tts_wav

---
 GPT_SoVITS/inference_webui.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py
index a361ed58..9e4a4407 100644
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@@ -765,8 +765,17 @@ def get_tts_wav(
     sample_steps=8,
     if_sr=False,
     pause_second=0.3,
+    device_override=None
 ):
     global cache
+    global device
+    if device_override:
+        device = device_override
+    
+    # Check if models are loaded
+    if (tokenizer is None or bert_model is None or ssl_model is None or 
+        vq_model is None or t2s_model is None):
+        raise RuntimeError("Models not loaded. Please call load_models() first.")
     if ref_wav_path:
         pass
     else: