Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add fairseq onnx support and strict configuration, fixes some onnx errors #2831

Merged
merged 1 commit into from
Aug 4, 2023

Conversation

SystemPanic
Copy link
Contributor

  • Adds fairseq models onnx support.
  • Make load_state_dict strict option configurable for customized fairseq models.
  • Fixes random out of index errors in onnx export caused by torch.randint generating char ids higher than model vocab size.
  • Fixes onnx export error when there is only a single speaker in a multi-speaker model.
  • Fixes onnx inference input mismatch when onnx exportation stripped out unused input arguments. Makes the input arguments variable instead of fixed, based on the real inputs used by the model.

Tested on more than 75 models, with multi-speaker and multi-language models, and single speaker and single language ones.

@erogol
Copy link
Member

erogol commented Aug 4, 2023

Great PR thanks.

I've come across a couple of exceptional pull requests from you recently. If you're interested, I can invite you to join the contributor's channel on Discord, where you can be a part of the development discussions please share your username, and I'll get you onboarded.

@erogol erogol merged commit 4e7f8cd into coqui-ai:dev Aug 4, 2023
44 checks passed
@SystemPanic SystemPanic deleted the onnxmultilang2 branch August 4, 2023 15:59
Tindell pushed a commit to pugtech-co/TTS that referenced this pull request Sep 4, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants