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 device flag to TTS CLI #2875

Merged
merged 1 commit into from
Aug 28, 2023
Merged

Add device flag to TTS CLI #2875

merged 1 commit into from
Aug 28, 2023

Conversation

jaketae
Copy link
Contributor

@jaketae jaketae commented Aug 14, 2023

Context

In #2855, we added a new .to(device) API for the TTS class, allowing users to freely choose the accelerator backend of their choice. For further ease of access, we want to incorporate this new feature into the TTS CLI, which currently only allows users to specify the device via --use_cuda.

Solution

Add a new --device flag to the TTS CLI. Now, uses can do

$ tts --text "hello world" --device cuda:1  # specify cuda device
$ tts --text "hello world" --device mps  # use apple silicon

@jaketae jaketae marked this pull request as ready for review August 14, 2023 22:31
@jaketae
Copy link
Contributor Author

jaketae commented Aug 14, 2023

Hi @erogol, this is a quick follow-up to the previous PR. In this PR, we add a --device flag to the TTS CLI. Let me know what you think! Thank you.

@erogol erogol merged commit b79b6f0 into coqui-ai:dev Aug 28, 2023
44 checks passed
@jaketae jaketae deleted the cli-device branch August 28, 2023 17:11
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