[Fix]: fix turbomind (#81)

fix turbomind
InternLM · Jan 16, 2024 · 6b28760 · 6b28760
1 parent 987618c
commit 6b28760
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lagent/llms/lmdeploy.py b/lagent/llms/lmdeploy.py
@@ -105,7 +105,11 @@ def __init__(self,
                  meta_template=None,
                  tp=1,
                  **kwargs):
-        super().__init__(path, max_seq_len, tokenizer_only, meta_template)
+        super().__init__(
+            path=path,
+            max_seq_len=max_seq_len,
+            tokenizer_only=tokenizer_only,
+            meta_template=meta_template)
         tokenizer_model_path = osp.join(path, 'triton_models', 'tokenizer')
         self.tokenizer = Tokenizer(tokenizer_model_path)
         self.tm_model = tm.TurboMind(