Skip to content

Commit 4bd1c67

Browse files
committed
add paligemma (1,2 and 2 mix)
1 parent b376399 commit 4bd1c67

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/llamafactory/model/model_utils/liger_kernel.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,8 @@ def apply_liger_kernel(
7878
from liger_kernel.transformers import apply_liger_kernel_to_gemma3 as apply_liger_kernel
7979
elif model_type == "gemma3_text":
8080
from liger_kernel.transformers import apply_liger_kernel_to_gemma3_text as apply_liger_kernel
81+
if model_type == "paligemma":
82+
from liger_kernel.transformers import apply_liger_kernel_to_paligemma as apply_liger_kernel
8183
elif model_type == "llama":
8284
from liger_kernel.transformers import apply_liger_kernel_to_llama as apply_liger_kernel
8385
elif model_type == "mistral":

0 commit comments

Comments
 (0)