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