ForwardTacotron icon indicating copy to clipboard operation
ForwardTacotron copied to clipboard
verified publisher icon as-ideas

How to save model using torch.jit to run generate on cpu without external code?

Open psydok opened this issue 4 years ago • 4 comments

psydok avatar Jun 03 '21 17:06 psydok

Hi, for torchscript it is necessary to do some changes to the model. I will investigate this if i have some spare time.

cschaefer26 avatar Jun 07 '21 12:06 cschaefer26

Any news on this?

faranaziz avatar Jun 21 '21 08:06 faranaziz

Hi, the current state is that I implemented a jit-compatible model here: https://github.com/as-ideas/ForwardTacotron/blob/experiments/enable_jit/models/forward_tacotron.py

It is taking a bit experimenting though until I decide to merge it to master, feel free to use the branch if you like!

cschaefer26 avatar Jun 22 '21 09:06 cschaefer26

Hi, good news, the jit export is implemented now: https://github.com/as-ideas/ForwardTacotron#export-model-with-torchscript

cschaefer26 avatar Jul 16 '21 09:07 cschaefer26