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

Amphion VALL-E new version release #220

Merged
merged 22 commits into from
Jun 21, 2024
Merged

Conversation

jiaqili3
Copy link
Collaborator

@jiaqili3 jiaqili3 commented Jun 17, 2024

✨ Description

In this PR, we release an unofficial PyTorch implementation of VALL-E, a zero-shot voice cloning model via neural codec language modeling. If trained properly, this model could match the performance specified in the original paper.
This is a refined version compared to the first version of VALLE in Amphion, we have changed the underlying implementation to Llama to provide better model performance, faster training speed, and more readable codes.
This can be a great tool for users who want to learn speech language models and its implementation.

🚧 Related Issues

None

👨‍💻 Changes Proposed

    • We have changed the underlying implementation to Llama to provide better model performance, faster training speed, and more readable codes.
    • We provide more detailed README.md for reproducing our models with pretrained weights, training on LibriTTS, and future plans on improving the model.
    • We use a refined codec model name SpeechTokenizer as the codec, yielding better modeling quality than the original Encodec

🧑‍🤝‍🧑 Who Can Review?

@HeCheng0625 @RMSnow @HarryHe11 @zhizhengwu

✅ Checklist

    • Code has been reviewed
    • Code complies with the project's code standards and best practices
    • Code has passed all tests
    • Code does not affect the normal use of existing features
    • Code has been commented properly
    • Documentation has been updated (if applicable)
    • Demo/checkpoint has been attached (if applicable)

Copy link
Collaborator

@RMSnow RMSnow left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks Jiaqi for the great efforts! I found a lot of codes are borrowed from speechtokenizer. You can add an acknowledge in the main readme.

README.md Outdated Show resolved Hide resolved
bins/tts/train.py Outdated Show resolved Hide resolved
egs/tts/valle_v2/README.md Show resolved Hide resolved
egs/tts/valle_v2/README.md Show resolved Hide resolved
egs/tts/valle_v2/train_ar_libritts.sh Outdated Show resolved Hide resolved
models/codec/speechtokenizer/quantization/core_vq.py Outdated Show resolved Hide resolved
models/codec/speechtokenizer/quantization/distrib.py Outdated Show resolved Hide resolved
models/codec/speechtokenizer/quantization/vq.py Outdated Show resolved Hide resolved
models/tts/valle_v2/modeling_llama.py Show resolved Hide resolved
models/tts/valle_v2/run_infer.py Outdated Show resolved Hide resolved
@RMSnow
Copy link
Collaborator

RMSnow commented Jun 17, 2024

Do we have any pretrained models or demo for this new valle?

@jiaqili3
Copy link
Collaborator Author

jiaqili3 commented Jun 18, 2024

Do we have any pretrained models or demo for this new valle?

It has been detailed in the readme file in egs/tts/valle_v2, and the demo.ipynb has also been uploaded to run inference with pretrained weights

@jiaqili3
Copy link
Collaborator Author

Hi @RMSnow , thanks for your review! I've updated the code and your previous review questions have been resolved.

@RMSnow RMSnow self-requested a review June 20, 2024 03:05
bins/tts/train.py Show resolved Hide resolved
egs/tts/valle_v2/demo.ipynb Outdated Show resolved Hide resolved
@RMSnow
Copy link
Collaborator

RMSnow commented Jun 20, 2024

Hi @jiaqili3, please update the demo.ipynb. Others look good to me.

@jiaqili3
Copy link
Collaborator Author

Updated. Thanks @RMSnow

@RMSnow RMSnow self-requested a review June 21, 2024 18:00
Copy link
Collaborator

@RMSnow RMSnow left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@RMSnow RMSnow changed the title Amphion VALLE new version release Amphion VALL-E new version release Jun 21, 2024
@RMSnow RMSnow merged commit f96a153 into open-mmlab:main Jun 21, 2024
1 check passed
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