search

NTT123 / vietTTS

Vietnamese Text to Speech library
MIT License
200 stars 91 forks source link

Prediction time is too slow #32

Open GitVHV opened 1 year ago

GitVHV commented 1 year ago

I see an issue coming from text2mel using only 1 cpu and mel2wave being all cpu. Do you have any solution to this problem to optimize processing time? Thank you so much!

Approximetal commented 1 year ago

Reason see this issue https://github.com/google/jax/issues/2521 I use pytorch version hifigan instead of haiku+jax, it is much faster than previous. I'm trying to fix text2mel part to solve the recompilation issue, but not finished yet.

GitVHV commented 1 year ago

Reason see this issue google/jax#2521 I use pytorch version hifigan instead of haiku+jax, it is much faster than previous. I'm trying to fix text2mel part to solve the recompilation issue, but not finished yet.

I also tried running with pytorch but it seems that the input of text2mel doesn't match. Let me know when you get a handle on it. Looking forward to your reply.