BUG: Fix auto recover kwargs (#885)

xorbitsai · Jan 11, 2024 · 0b3f6f7 · 0b3f6f7
1 parent 6bfd80f
commit 0b3f6f7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xinference/core/worker.py b/xinference/core/worker.py
@@ -339,6 +339,8 @@ async def launch_builtin_model(
     ):
         launch_args = locals()
         launch_args.pop("self")
+        launch_args.pop("kwargs")
+        launch_args.update(kwargs)
         if n_gpu is not None:
             if isinstance(n_gpu, int) and (n_gpu <= 0 or n_gpu > cuda_count()):
                 raise ValueError(