BUG: 🐛 fix unable launch qwen2-embedding (#2185)

Co-authored-by: 胡子俊 <[email protected]>
xorbitsai · Aug 29, 2024 · de80532 · de80532
1 parent b91aae8
commit de80532
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xinference/model/embedding/core.py b/xinference/model/embedding/core.py
@@ -124,13 +124,15 @@ def __init__(
         model_path: str,
         model_spec: EmbeddingModelSpec,
         device: Optional[str] = None,
+        **kwargs,
     ):
         self._model_uid = model_uid
         self._model_path = model_path
         self._device = device
         self._model = None
         self._counter = 0
         self._model_spec = model_spec
+        self._kwargs = kwargs
 
     def load(self):
         try: