facebookresearch / fairseq

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
MIT License
30.16k stars 6.37k forks source link

Is it possible to finetune the MMS model? #5173

Open CopyNinja1999 opened 1 year ago

CopyNinja1999 commented 1 year ago

Is it possible to finetune the MMS model? I didn't find any doc or code supporting this feature.

patrickvonplaten commented 1 year ago

We're working on making this very easy in transformers. You can check:

Stay tuned!

CopyNinja1999 commented 1 year ago

@patrickvonplaten Hi, could you please also upload the pretrained discriminator for mms-tts models? Also, I met this error when fine-tuning the Korean tts model using random discriminator checkpoint: packages/bitsandbytes/optim/optimizer.py", line 455, in update_step if state["state1"].dtype == torch.float: KeyError: 'state1'

allandclive commented 1 year ago

https://github.com/huggingface/transformers/tree/main/examples/pytorch/speech-recognition#connectionist-temporal-classification-with-adapters

patrickvonplaten commented 1 year ago

Also a more step-by-step blog post here: https://huggingface.co/blog/mms_adapters