Fix Whisper formatting

This commit is contained in:
Mark Backman
2025-03-29 17:57:50 -04:00
parent b9ea3f0fd9
commit a978a5cd4a

View File

@@ -9,10 +9,10 @@
import asyncio import asyncio
from enum import Enum from enum import Enum
from typing import AsyncGenerator, Optional from typing import AsyncGenerator, Optional
from typing_extensions import TYPE_CHECKING, override
import numpy as np import numpy as np
from loguru import logger from loguru import logger
from typing_extensions import TYPE_CHECKING, override
from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
from pipecat.services.ai_services import SegmentedSTTService from pipecat.services.ai_services import SegmentedSTTService
@@ -332,6 +332,7 @@ class WhisperSTTService(SegmentedSTTService):
""" """
try: try:
from faster_whisper import WhisperModel from faster_whisper import WhisperModel
logger.debug("Loading Whisper model...") logger.debug("Loading Whisper model...")
self._model = WhisperModel( self._model = WhisperModel(
self.model_name, device=self._device, compute_type=self._compute_type self.model_name, device=self._device, compute_type=self._compute_type
@@ -456,10 +457,11 @@ class WhisperSTTServiceMLX(WhisperSTTService):
whisper_lang = self.language_to_service_language(self._settings["language"]) whisper_lang = self.language_to_service_language(self._settings["language"])
chunk = await asyncio.to_thread( chunk = await asyncio.to_thread(
mlx_whisper.transcribe, audio_float, mlx_whisper.transcribe,
audio_float,
path_or_hf_repo=self.model_name, path_or_hf_repo=self.model_name,
temperature=self._temperature, temperature=self._temperature,
language=whisper_lang language=whisper_lang,
) )
text: str = "" text: str = ""
for segment in chunk.get("segments", []): for segment in chunk.get("segments", []):