chore: rewrite .cnb.yml to CI pipeline format with api_trigger event

This commit is contained in:
shuiyu1123 committed 2026-06-18 18:27:38 +00:00
1 parent 9799aa00c5
commit edf3a83e3c
1 file changed
+16 -15
+16 -15
View File
@@ -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
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