blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-08 00:06:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update export_torch_script.py (#2494)

Browse Source 
Avoid dtype inconsistency when exporting
This commit is contained in:
Yixiao Chen 2025-07-02 10:48:28 -04:00 committed by GitHub
parent 6df61f58e4
commit 8c579d46dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/export_torch_script.py
View File

@ -474,6 +474,10 @@ class T2SModel(nn.Module):
bert = bert.unsqueeze(0) bert = bert.unsqueeze(0)
x = self.ar_text_embedding(all_phoneme_ids) x = self.ar_text_embedding(all_phoneme_ids)
# avoid dtype inconsistency when exporting
bert = bert.to(dtype=self.bert_proj.weight.dtype)
x = x + self.bert_proj(bert.transpose(1, 2)) x = x + self.bert_proj(bert.transpose(1, 2))
x: torch.Tensor = self.ar_text_position(x) x: torch.Tensor = self.ar_text_position(x)