fix mistake

WAYKEN-TSE · Jul 28, 2023 · 2e66373 · 2e66373
1 parent 4806eec
commit 2e66373
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlevlp/examples/blip2/run_pretrain_stage2.py b/paddlevlp/examples/blip2/run_pretrain_stage2.py
@@ -247,7 +247,7 @@ def main():
     logger.info("training_args.use_hybrid_parallel:{}".format(
         training_args.use_hybrid_parallel))
     # create trainer
-    load_pretrained_model(model, LLM_LIST[model_args.llm_name])
+    load_pretrained_model(model.language_model, LLM_LIST[model_args.llm_name])
     load_pretrained_model(model, training_args.pretrained_model_path)
     trainer = Trainer(
         model=model,