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()