support train from scratch #4033 #4075

This commit is contained in:
hiyouga committed 2024-06-06 02:43:19 +08:00
1 parent 946f601136
commit a12a506c3d
2 files changed
+6

No files matched your search

+2
View File
@@ -131,6 +131,8 @@ def load_model(
model = load_mod_pretrained_model(**init_kwargs)
elif model_args.visual_inputs:
model = AutoModelForVision2Seq.from_pretrained(**init_kwargs)
elif model_args.train_from_scratch:
model = AutoModelForCausalLM.from_config(config)
else:
model = AutoModelForCausalLM.from_pretrained(**init_kwargs)