[model] add kimi vl 2506 (#8432)

2025-06-23 17:56:48 +08:00
parent 7b252b2368
commit 9af7915f7b
4 changed files with 36 additions and 30 deletions
--- a/src/llamafactory/model/model_utils/unsloth.py
+++ b/src/llamafactory/model/model_utils/unsloth.py
@@ -80,7 +80,10 @@ def get_unsloth_peft_model(


 def load_unsloth_peft_model(
-    config: "PretrainedConfig", model_args: "ModelArguments", finetuning_args: "FinetuningArguments", is_trainable: bool
+    config: "PretrainedConfig",
+    model_args: "ModelArguments",
+    finetuning_args: "FinetuningArguments",
+    is_trainable: bool,
 ) -> "PreTrainedModel":
    r"""Load peft model with unsloth. Used in both training and inference."""
    from unsloth import FastLanguageModel  # type: ignore