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

services(fal,moondream): add missing **kwargs

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-09-18 22:48:06 -07:00
parent 0e8f56c752
commit 3298f935ef
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/pipecat/services/fal.py
View File

@@ -43,8 +43,9 @@ class FalImageGenService(ImageGenService):
aiohttp_session: aiohttp.ClientSession, aiohttp_session: aiohttp.ClientSession,
model: str = "fal-ai/fast-sdxl", model: str = "fal-ai/fast-sdxl",
key: str | None = None, key: str | None = None,
**kwargs
): ):
super().__init__() super().__init__(**kwargs)
self._model = model self._model = model
self._params = params self._params = params
self._aiohttp_session = aiohttp_session self._aiohttp_session = aiohttp_session

7
src/pipecat/services/moondream.py
View File

@@ -46,12 +46,13 @@ def detect_device():
class MoondreamService(VisionService): class MoondreamService(VisionService):
def __init__( def __init__(
self, self,
*, *,
model="vikhyatk/moondream2", model="vikhyatk/moondream2",
revision="2024-08-26", revision="2024-08-26",
use_cpu=False use_cpu=False,
**kwargs
): ):
super().__init__() super().__init__(**kwargs)
if not use_cpu: if not use_cpu:
device, dtype = detect_device() device, dtype = detect_device()