From b8ce03fd1bf30d5c0072d6a1946baf028eeab846 Mon Sep 17 00:00:00 2001 From: chasonjiang <1440499136@qq.com> Date: Thu, 14 Mar 2024 11:24:10 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=81=A5=E5=A3=AE=E6=80=A7?= =?UTF-8?q?=EF=BC=8C=E9=98=B2=E6=AD=A2=E5=9C=A8cpu=E6=8E=A8=E7=90=86?= =?UTF-8?q?=E6=97=B6=E8=AE=BE=E7=BD=AE=E5=8D=8A=E7=B2=BE=E5=BA=A6=E6=8A=A5?= =?UTF-8?q?=E9=94=99?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- GPT_SoVITS/TTS_infer_pack/TTS.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/GPT_SoVITS/TTS_infer_pack/TTS.py b/GPT_SoVITS/TTS_infer_pack/TTS.py index e1ed8bb..d82ed5d 100644 --- a/GPT_SoVITS/TTS_infer_pack/TTS.py +++ b/GPT_SoVITS/TTS_infer_pack/TTS.py @@ -228,7 +228,7 @@ class TTS: self.cnhuhbert_model = CNHubert(base_path) self.cnhuhbert_model=self.cnhuhbert_model.eval() self.cnhuhbert_model = self.cnhuhbert_model.to(self.configs.device) - if self.configs.is_half: + if self.configs.is_half and str(self.configs.device)!="cpu": self.cnhuhbert_model = self.cnhuhbert_model.half() @@ -239,7 +239,7 @@ class TTS: self.bert_model = AutoModelForMaskedLM.from_pretrained(base_path) self.bert_model=self.bert_model.eval() self.bert_model = self.bert_model.to(self.configs.device) - if self.configs.is_half: + if self.configs.is_half and str(self.configs.device)!="cpu": self.bert_model = self.bert_model.half() @@ -272,7 +272,7 @@ class TTS: vits_model = vits_model.eval() vits_model.load_state_dict(dict_s2["weight"], strict=False) self.vits_model = vits_model - if self.configs.is_half: + if self.configs.is_half and str(self.configs.device)!="cpu": self.vits_model = self.vits_model.half() @@ -290,7 +290,7 @@ class TTS: t2s_model = t2s_model.to(self.configs.device) t2s_model = t2s_model.eval() self.t2s_model = t2s_model - if self.configs.is_half: + if self.configs.is_half and str(self.configs.device)!="cpu": self.t2s_model = self.t2s_model.half() def enable_half_precision(self, enable: bool = True): @@ -300,7 +300,7 @@ class TTS: enable: bool, whether to enable half precision. ''' - if self.configs.device == "cpu" and enable: + if str(self.configs.device) == "cpu" and enable: print("Half precision is not supported on CPU.") return