diff --git a/examples/pytorch/stable_diffusion/cones2/finetune_stable_diffusion_cones2.py b/examples/pytorch/stable_diffusion/cones2/finetune_stable_diffusion_cones2.py
index de97b3c9..135a5c7d 100644
--- a/examples/pytorch/stable_diffusion/cones2/finetune_stable_diffusion_cones2.py
+++ b/examples/pytorch/stable_diffusion/cones2/finetune_stable_diffusion_cones2.py
@@ -77,7 +77,7 @@ def cfg_modify_fn(cfg):
 
 
 kwargs = dict(
-    model=training_args.model_id,
+    model=training_args.model,
     model_revision=args.model_revision,
     work_dir=training_args.work_dir,
     train_dataset=train_dataset,