From c00aec65bc29edf585084e0a6713f023ac73c0e8 Mon Sep 17 00:00:00 2001
From: tianchongchong <58495634+tianchongchong@users.noreply.github.com>
Date: Fri, 20 Dec 2024 14:50:41 +0800
Subject: [PATCH] Update train_olmo.ipynb
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Change positional args to keyword args，because the second positional arg is dataloader.
---
 examples/train_olmo.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)"