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