diff --git a/xinference/model/embedding/core.py b/xinference/model/embedding/core.py index 23a9766c48..7ad985ff09 100644 --- a/xinference/model/embedding/core.py +++ b/xinference/model/embedding/core.py @@ -124,6 +124,7 @@ def __init__( model_path: str, model_spec: EmbeddingModelSpec, device: Optional[str] = None, + **kwargs, ): self._model_uid = model_uid self._model_path = model_path @@ -131,6 +132,7 @@ def __init__( self._model = None self._counter = 0 self._model_spec = model_spec + self._kwargs = kwargs def load(self): try: