From dfe1bd41d8534ee2532950107821c4547a33426b Mon Sep 17 00:00:00 2001
From: Arcitec <38923130+Arcitec@users.noreply.github.com>
Date: Mon, 8 Sep 2025 16:17:17 +0200
Subject: [PATCH] refactor: Remove code duplication in DeepSpeed dependency
 check

---
 indextts/infer_v2.py | 24 ++++++++----------------
 1 file changed, 8 insertions(+), 16 deletions(-)

diff --git a/indextts/infer_v2.py b/indextts/infer_v2.py
index 4d53e99..2f0f1d3 100644
--- a/indextts/infer_v2.py
+++ b/indextts/infer_v2.py
@@ -82,23 +82,15 @@ class IndexTTS2:
         else:
             self.gpt.eval()
         print(">> GPT weights restored from:", self.gpt_path)
-        if self.is_fp16:
-            use_deepspeed = True
-            try:
-                import deepspeed
-            except (ImportError, OSError, CalledProcessError) as e:
-                use_deepspeed = False
-                print(f">> DeepSpeed加载失败，回退到标准推理: {e}")
 
-            self.gpt.post_init_gpt2_config(use_deepspeed=use_deepspeed, kv_cache=True, half=True)
-        else:
-            use_deepspeed = True
-            try:
-                import deepspeed
-            except (ImportError, OSError, CalledProcessError) as e:
-                use_deepspeed = False
-                print(f">> DeepSpeed加载失败，回退到标准推理: {e}")
-            self.gpt.post_init_gpt2_config(use_deepspeed=use_deepspeed, kv_cache=True, half=False)
+        use_deepspeed = True
+        try:
+            import deepspeed
+        except (ImportError, OSError, CalledProcessError) as e:
+            use_deepspeed = False
+            print(f">> DeepSpeed加载失败，回退到标准推理: {e}")
+
+        self.gpt.post_init_gpt2_config(use_deepspeed=use_deepspeed, kv_cache=True, half=self.is_fp16)
 
         if self.use_cuda_kernel:
             # preload the CUDA kernel for BigVGAN