feat: add CNB cloud training config (GPU Dockerfile + .cnb.yml + train_cnb.py)

This commit is contained in:
shuiyu1123 committed 2026-06-18 18:09:59 +00:00
1 parent eb7717e9ce
commit 456c443e9c
4 files changed
+223 -1

No files matched your search

+15
View File
@@ -0,0 +1,15 @@
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 50 --batch_size 16