diff --git a/.cnb.yml b/.cnb.yml
index e7f2905..18912ae 100644
--- a/.cnb.yml
+++ b/.cnb.yml
@@ -1,15 +1,16 @@
-version: "3"
-tasks:
-  - name: "训练小狸AI"
-    image:
-      dockerfile: .ide/Dockerfile
-    resources:
-      cpu: 8
-      memory: 32G
-      gpu: 1
-    env:
-      CUDA_VISIBLE_DEVICES: "0"
-      PYTHONUNBUFFERED: "1"
-    commands:
-      - cd /workspace && git clone https://cnb.cool/ETeam_sy/xiaoli-llm-chat.git
-      - cd xiaoli-llm-chat && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
\ No newline at end of file
+master:
+  api_trigger:
+    - name: 训练小狸AI-Epoch100
+      docker:
+        build: .ide/Dockerfile
+      runner:
+        cpus: 8
+        tags: cnb:arch:amd64:gpu
+      env:
+        CUDA_VISIBLE_DEVICES: "0"
+        PYTHONUNBUFFERED: "1"
+      stages:
+        - name: train
+          script: |
+            cd /workspace && git clone https://cnb.cool/ETeam_sy/xiaoli-llm-chat.git
+            cd xiaoli-llm-chat && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
\ No newline at end of file