This commit is contained in:
hoshi-hiyouga authored and GitHub committed 2024-10-23 20:49:44 +08:00
1 parent 74a79cc059
commit b4c7dd3ac5
1 file changed
+2 -1
+2 -1
View File
@@ -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)