fix #5797

hiyouga · Oct 23, 2024 · b4c7dd3 · b4c7dd3
1 parent 74a79cc
commit b4c7dd3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)