diff --git a/src/llamafactory/model/model_utils/liger_kernel.py b/src/llamafactory/model/model_utils/liger_kernel.py index 2cf30b11..72c01f6f 100644 --- a/src/llamafactory/model/model_utils/liger_kernel.py +++ b/src/llamafactory/model/model_utils/liger_kernel.py @@ -47,6 +47,8 @@ def apply_liger_kernel( from liger_kernel.transformers import apply_liger_kernel_to_gemma3_text as apply_liger_kernel elif model_type == "glm4": from liger_kernel.transformers import apply_liger_kernel_to_glm4 as apply_liger_kernel + elif model_type == "glm4v": + from liger_kernel.transformers import apply_liger_kernel_to_glm4v as apply_liger_kernel elif model_type == "granite": from liger_kernel.transformers import apply_liger_kernel_to_granite as apply_liger_kernel elif model_type == "llama":