新增gpu模式下chattts代码

Chat2TTS/core.py

@@ -100,8 +100,17 @@ class Chat:
             assert gpt_ckpt_path, 'gpt_ckpt_path should not be None'
             gpt.load_state_dict(torch.load(gpt_ckpt_path, map_location='cpu'))
             self.pretrain_models['gpt'] = gpt
+            self.gpt = gpt
             self.logger.log(logging.INFO, 'gpt loaded.')
-            
+            spk_stat_path = os.path.join(os.path.dirname(gpt_ckpt_path), "spk_stat.pt")
+            assert os.path.exists(
+                spk_stat_path
+            ), f"Missing spk_stat.pt: {spk_stat_path}"
+            self.pretrain_models["spk_stat"] = torch.load(
+                spk_stat_path, weights_only=True, mmap=True, map_location='cpu'
+            ).to(device)
+
+
         if decoder_config_path:
             cfg = OmegaConf.load(decoder_config_path)
             decoder = DVAE(**cfg).to(device).eval()