diff --git a/src/pipecat/services/fal.py b/src/pipecat/services/fal.py index 4d99f6066..672135d02 100644 --- a/src/pipecat/services/fal.py +++ b/src/pipecat/services/fal.py @@ -43,8 +43,9 @@ class FalImageGenService(ImageGenService): aiohttp_session: aiohttp.ClientSession, model: str = "fal-ai/fast-sdxl", key: str | None = None, + **kwargs ): - super().__init__() + super().__init__(**kwargs) self._model = model self._params = params self._aiohttp_session = aiohttp_session diff --git a/src/pipecat/services/moondream.py b/src/pipecat/services/moondream.py index 10ac3353e..3441aeeb9 100644 --- a/src/pipecat/services/moondream.py +++ b/src/pipecat/services/moondream.py @@ -46,12 +46,13 @@ def detect_device(): class MoondreamService(VisionService): def __init__( self, - *, + *, model="vikhyatk/moondream2", revision="2024-08-26", - use_cpu=False + use_cpu=False, + **kwargs ): - super().__init__() + super().__init__(**kwargs) if not use_cpu: device, dtype = detect_device()