fix #5797

Former-commit-id: 71d23ed3444f24b31785d9f0f6dd711f6f516731
2024-10-23 20:49:44 +08:00
parent 3a8b2890eb
commit 6a2cd129c0
1 changed files with 2 additions and 1 deletions
--- a/src/llamafactory/model/loader.py
+++ b/src/llamafactory/model/loader.py
@@ -153,8 +153,9 @@ def load_model(
                load_class = AutoModelForVision2Seq
            else:
                load_class = AutoModelForCausalLM
+
            if model_args.train_from_scratch:
-                model = load_class.from_config(config)
+                model = load_class.from_config(config, trust_remote_code=True)
            else:
                model = load_class.from_pretrained(**init_kwargs)