diff --git a/examples/train_olmo.ipynb b/examples/train_olmo.ipynb
index e8a7ed2..4419c2d 100644
--- a/examples/train_olmo.ipynb
+++ b/examples/train_olmo.ipynb
@@ -256,7 +256,7 @@
     "config.dist.fsdp.wrap_layer_cls = {\"OlmoSequentialBlock\"} # 传入将OLMo模型的decoder layer进行FSDP封装\n",
     "\n",
     "# 一行代码加速模型\n",
-    "model = torchacc.accelerate(model, config)\n",
+    "model = torchacc.accelerate(model=model, config=config)\n",
     "\n",
     "# 异步加速数据加载\n",
     "train_loader = torchacc.AsyncLoader(train_loader, model.device)"