wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

services(riva): model -> function_id

Browse Source
This commit is contained in:
vipyne
2024-12-09 20:57:16 -06:00
parent 5a98ae6380
commit 3f9d39329c
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/pipecat/services/riva.py
View File

@@ -48,8 +48,7 @@ class FastpitchTTSService(TTSService):
server: str = "grpc.nvcf.nvidia.com:443", server: str = "grpc.nvcf.nvidia.com:443",
voice_id: str = "English-US.Female-1", voice_id: str = "English-US.Female-1",
sample_rate: int = 24000, sample_rate: int = 24000,
# nvidia riva calls this 'function-id' function_id: str = "0149dedb-2be8-4195-b9a0-e57e0e14f972",
model: str = "0149dedb-2be8-4195-b9a0-e57e0e14f972",
params: InputParams = InputParams(), params: InputParams = InputParams(),
**kwargs, **kwargs,
): ):
@@ -64,7 +63,7 @@ class FastpitchTTSService(TTSService):
self.set_voice(voice_id) self.set_voice(voice_id)
metadata = [ metadata = [
["function-id", model], ["function-id", function_id],
["authorization", f"Bearer {api_key}"], ["authorization", f"Bearer {api_key}"],
] ]
auth = riva.client.Auth(None, True, server, metadata) auth = riva.client.Auth(None, True, server, metadata)
@@ -118,8 +117,7 @@ class ParakeetSTTService(STTService):
*, *,
api_key: str, api_key: str,
server: str = "grpc.nvcf.nvidia.com:443", server: str = "grpc.nvcf.nvidia.com:443",
# nvidia calls this 'function-id' function_id: str = "1598d209-5e27-4d3c-8079-4751568b1081",
model: str = "1598d209-5e27-4d3c-8079-4751568b1081",
params: InputParams = InputParams(), params: InputParams = InputParams(),
**kwargs, **kwargs,
): ):
@@ -143,7 +141,7 @@ class ParakeetSTTService(STTService):
self.set_model_name("parakeet-ctc-1.1b-asr") self.set_model_name("parakeet-ctc-1.1b-asr")
metadata = [ metadata = [
["function-id", model], ["function-id", function_id],
["authorization", f"Bearer {api_key}"], ["authorization", f"Bearer {api_key}"],
] ]
auth = riva.client.Auth(None, True, server, metadata) auth = riva.client.Auth(None, True, server, metadata)