EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Fix running MeloTTS models on GPU. (#1379)

Browse Source 
We need to use opset 18 to export the model to onnx.
This commit is contained in:
Fangjun Kuang
2024-09-26 16:51:43 +08:00
committed by GitHub
parent 69c8e7b9d0
commit 12d04ce8ed
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/melo-tts/export-onnx.py
View File

@@ -229,7 +229,7 @@ def main():
torch_model = ModelWrapper(model) torch_model = ModelWrapper(model)
opset_version = 13 opset_version = 18
x = torch.randint(low=0, high=10, size=(60,), dtype=torch.int64) x = torch.randint(low=0, high=10, size=(60,), dtype=torch.int64)
print(x.shape) print(x.shape)
x_lengths = torch.tensor([x.size(0)], dtype=torch.int64) x_lengths = torch.tensor([x.size(0)], dtype=torch.int64)