Add missing key to init_kwargs

hiyouga · Dec 12, 2024 · 2f41407 · 2f41407
1 parent 514fcf6
commit 2f41407
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/llamafactory/model/loader.py b/src/llamafactory/model/loader.py
@@ -157,6 +157,7 @@ def load_model(
             if model_args.train_from_scratch:
                 model = load_class.from_config(config, trust_remote_code=model_args.trust_remote_code)
             else:
+                init_kwargs["trust_remote_code"] = model_args.trust_remote_code
                 model = load_class.from_pretrained(**init_kwargs)
 
         if model_args.mixture_of_depths == "convert":