fix #5797
This commit is contained in:
1 parent
74a79cc059
commit
b4c7dd3ac5
1 file changed
+2
-1
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user