services(fal,moondream): add missing **kwargs
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user