huawei-noah / Speech-Backbones

This is the main repository of open-sourced speech technology by Huawei Noah's Ark Lab.
545 stars 113 forks source link

GradTTS device compatibility #34

Closed bukhalmae145 closed 8 months ago

bukhalmae145 commented 8 months ago

I tried to use GradTTS by moving the model to MPS (.to('mps')) but the terminal's error message told me that MPS doesn't support 3D padding. So I tried it on CPU and it worked but it is too slow. Is there any way to adapt GradTTS to Mac's MPS device?