From 1572a3f6b3a93a9e55fc65d51a2e2d617551536f Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Mon, 16 Dec 2024 15:29:56 +0800 Subject: [PATCH] fix --- swift/llm/train/tuner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/swift/llm/train/tuner.py b/swift/llm/train/tuner.py index 750223caff..3b9a65bb76 100644 --- a/swift/llm/train/tuner.py +++ b/swift/llm/train/tuner.py @@ -28,9 +28,9 @@ def apply_liger(model_type: str): apply_liger_kernel_to_mistral() elif model_type in (ModelType.mixtral): apply_liger_kernel_to_mixtral() - elif model_type in (ModelType.gemma): + elif model_type in (ModelType.gemma, ModelType.gemma2): apply_liger_kernel_to_gemma() - elif model_type in (ModelType.gemma2): + elif model_type in (ModelType.qwen2, ModelType.qwen2_5): apply_liger_kernel_to_qwen2() elif model_type in (ModelType.phi3): apply_liger_kernel_to_phi3()