mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 03:32:50 +08:00
[misc] enable liger kernel for gemma3 (#7462)
This commit is contained in:
parent
9abee9cd1a
commit
59a56f7226
@ -74,6 +74,8 @@ def apply_liger_kernel(
|
||||
from liger_kernel.transformers import apply_liger_kernel_to_gemma as apply_liger_kernel
|
||||
elif model_type == "gemma2":
|
||||
from liger_kernel.transformers import apply_liger_kernel_to_gemma2 as apply_liger_kernel
|
||||
elif model_type == "gemma3":
|
||||
from liger_kernel.transformers import apply_liger_kernel_to_gemma3 as apply_liger_kernel
|
||||
elif model_type == "llama":
|
||||
from liger_kernel.transformers import apply_liger_kernel_to_llama as apply_liger_kernel
|
||||
elif model_type == "mistral":
|
||||
|
Loading…
x
Reference in New Issue
Block a user