Updated Smart Turn model weights to v3.2
This commit is contained in:
@@ -162,7 +162,7 @@ where = ["src"]
|
|||||||
"src/pipecat/audio/dtmf/dtmf-star.wav",
|
"src/pipecat/audio/dtmf/dtmf-star.wav",
|
||||||
]
|
]
|
||||||
"pipecat.services.aws_nova_sonic" = ["src/pipecat/services/aws_nova_sonic/ready.wav"]
|
"pipecat.services.aws_nova_sonic" = ["src/pipecat/services/aws_nova_sonic/ready.wav"]
|
||||||
"pipecat.audio.turn.smart_turn.data" = ["src/pipecat/audio/turn/smart_turn/data/smart-turn-v3.1-cpu.onnx"]
|
"pipecat.audio.turn.smart_turn.data" = ["src/pipecat/audio/turn/smart_turn/data/smart-turn-v3.2-cpu.onnx"]
|
||||||
|
|
||||||
[tool.pytest.ini_options]
|
[tool.pytest.ini_options]
|
||||||
addopts = "--verbose"
|
addopts = "--verbose"
|
||||||
|
|||||||
Binary file not shown.
@@ -42,7 +42,7 @@ class LocalSmartTurnAnalyzerV3(BaseSmartTurn):
|
|||||||
|
|
||||||
Args:
|
Args:
|
||||||
smart_turn_model_path: Path to the ONNX model file. If this is not
|
smart_turn_model_path: Path to the ONNX model file. If this is not
|
||||||
set, the bundled smart-turn-v3.1-cpu model will be used.
|
set, the bundled smart-turn-v3.2-cpu model will be used.
|
||||||
cpu_count: The number of CPUs to use for inference. Defaults to 1.
|
cpu_count: The number of CPUs to use for inference. Defaults to 1.
|
||||||
**kwargs: Additional arguments passed to BaseSmartTurn.
|
**kwargs: Additional arguments passed to BaseSmartTurn.
|
||||||
"""
|
"""
|
||||||
@@ -50,7 +50,7 @@ class LocalSmartTurnAnalyzerV3(BaseSmartTurn):
|
|||||||
|
|
||||||
if not smart_turn_model_path:
|
if not smart_turn_model_path:
|
||||||
# Load bundled model
|
# Load bundled model
|
||||||
model_name = "smart-turn-v3.1-cpu.onnx"
|
model_name = "smart-turn-v3.2-cpu.onnx"
|
||||||
package_path = "pipecat.audio.turn.smart_turn.data"
|
package_path = "pipecat.audio.turn.smart_turn.data"
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|||||||
Reference in New Issue
Block a user