Audio-to-Audio
Moshi
Safetensors
English

Do you provide the dataset schema for fine-tune the moshi RAG model?

#2
by gnomefin - opened

I am interested on observing this model, however I can't find the fine-tuning schema or script. Could you please release it atleast the dataset schema?

Kyutai org

The MoshiRAG paper lists the stats of the fine-tuning dataset in the appendix. As the audio data was created by Kyutai's internal conversational TTS model, the release of this dataset will be subject to Kyutai's plan of releasing the conversational TTS model. However, there are some open-sourced conversational TTS models available on HuggingFace which may be useful if you would like to re-construct the training set on your own.

As for the training recipe, I would encourage you to look into the moshi-finetune repo. Training of MoshiRAG follows a very similar pipeline. The only major difference is the added ARC-Encoder reference encoder -- the detailed model architecture is defined in this script.

Sign up or log in to comment