Fix Kwargs (#885)

deepjavalibrary · Jun 30, 2023 · 4bf965e · 4bf965e
1 parent cfb23d8
commit 4bf965e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engines/python/setup/djl_python/rolling_batch/scheduler_rolling_batch.py b/engines/python/setup/djl_python/rolling_batch/scheduler_rolling_batch.py
@@ -77,7 +77,7 @@ def preprocess_requests(self, requests):
 
     def _init_model_and_tokenizer(self, kwargs, model_id_or_path):
         self.config = AutoConfig.from_pretrained(model_id_or_path,
-                                                 kwargs=kwargs)
+                                                 **kwargs)
         architectures = self.config.architectures
         if architectures and architectures[0].endswith(
                 "ForConditionalGeneration"):