From ee840b4e01474cf6558ede28e0b079c9bc8238ba Mon Sep 17 00:00:00 2001 From: Dain Kim Date: Thu, 10 Apr 2025 12:27:30 +0900 Subject: [PATCH] [bugfix] enable_gemma_liger_kernel (#7660) - The `enable_liger_kernel` function for the Gemma model series was not executed due to the existing `if` statement in the code. - Changed the line to an `elif` statement so that the `apply_liger_kernel` function is executed properly. resolved: #7628 --- src/llamafactory/model/model_utils/liger_kernel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llamafactory/model/model_utils/liger_kernel.py b/src/llamafactory/model/model_utils/liger_kernel.py index 4f6cbf2f..486c54e6 100644 --- a/src/llamafactory/model/model_utils/liger_kernel.py +++ b/src/llamafactory/model/model_utils/liger_kernel.py @@ -45,7 +45,7 @@ def apply_liger_kernel( from liger_kernel.transformers import apply_liger_kernel_to_gemma3 as apply_liger_kernel elif model_type == "gemma3_text": from liger_kernel.transformers import apply_liger_kernel_to_gemma3_text as apply_liger_kernel - if model_type == "paligemma": + elif model_type == "paligemma": from liger_kernel.transformers import apply_liger_kernel_to_paligemma as apply_liger_kernel elif model_type == "llama": from liger_kernel.transformers import apply_liger_kernel_to_llama as apply_liger_kernel