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.

Sign up for GitHub

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

Enable GPU support #5

Open
wants to merge 12 commits into
base: master
Choose a base branch
from
Open

Enable GPU support #5

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
48de374
enable GPU support
Nov 12, 2023
e8f3317
Add wyoming-piper executable
mweinelt Oct 30, 2023
49e102a
Update README.md with local install
monkeyclass Oct 17, 2023
cafb52c
Add test
synesthesiam Feb 19, 2024
b4d6e59
Update voices.json
synesthesiam Feb 19, 2024
b76de59
Relax test requirements
synesthesiam Feb 19, 2024
7b6ec48
Lint
synesthesiam Feb 19, 2024
068dad3
Increase timeout for download
synesthesiam Feb 19, 2024
633ee62
Update dev requirements
synesthesiam Feb 26, 2024
0d9bcb0
Update changelog
synesthesiam Feb 26, 2024
4e5a557
change --cuda to --use-cuda
Jan 9, 2025
5f45edf
fix use cuda argument
mreilaender Jan 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions wyoming_piper/__main__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,11 @@ async def main() -> None:
action="store_true",
help="Download latest voices.json during startup",
)
parser.add_argument(
"--cuda",
action="store_true",
help="Use GPU"
)
#
parser.add_argument("--debug", action="store_true", help="Log DEBUG messages")
args = parser.parse_args()
Expand Down
3 changes: 3 additions & 0 deletions wyoming_piper/process.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -148,6 +148,9 @@ async def get_process(self, voice_name: Optional[str] = None) -> PiperProcess:
if self.args.noise_w:
piper_args.extend(["--noise-w", str(self.args.noise_w)])

if self.args.cuda:
piper_args.extend(["--cuda"])

_LOGGER.debug(
"Starting piper process: %s args=%s", self.args.piper, piper_args
)
Expand Down