fix

modelscope · Dec 24, 2024 · ea29e31 · ea29e31
1 parent 9d46266
commit ea29e31
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/swift/llm/infer/infer_engine/pt_engine.py b/swift/llm/infer/infer_engine/pt_engine.py
@@ -78,6 +78,7 @@ def __init__(
         for adapter in self.adapters:
             self._add_adapter(safe_snapshot_download(adapter, use_hf=use_hf, hub_token=hub_token))
         self._post_init()
+        self.task_type = 'causal_lm'
 
     def _post_init(self):
         super()._post_init()