From ea29e311a266ff9069e207dd6fe59fd66086e045 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Wed, 25 Dec 2024 00:34:23 +0800 Subject: [PATCH] fix --- swift/llm/infer/infer_engine/pt_engine.py | 1 + 1 file changed, 1 insertion(+) diff --git a/swift/llm/infer/infer_engine/pt_engine.py b/swift/llm/infer/infer_engine/pt_engine.py index 2dfe82d9e..978ce90bc 100644 --- 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()