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

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

View File

@@ -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

View File

@@ -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()