remove visual_inputs, fix qlora

Former-commit-id: be30c01c4f1482520ece770bd54c6a4837c26f0a
2024-08-31 00:24:51 +08:00
parent d789b667d7
commit 2f6fc27c8b
22 changed files with 112 additions and 106 deletions
--- a/src/llamafactory/model/patcher.py
+++ b/src/llamafactory/model/patcher.py
@@ -131,11 +131,9 @@ def patch_model(
    if model_args.resize_vocab:
        resize_embedding_layer(model, tokenizer)

-    if model_args.visual_inputs:
-        autocast_projector_dtype(model, model_args)
-
    if is_trainable:
        prepare_model_for_training(model, model_args)
+        autocast_projector_dtype(model, model_args)
        add_z3_leaf_module(model)

    if not model_args.use_unsloth: