| version: '3.8' | |
| services: | |
| axolotl: | |
| image: winglian/axolotl:main-py3.10-cu118-2.0.1 | |
| restart: on-failure | |
| command: tail -f /dev/null | |
| volumes: | |
| - ./config:/workspace/axolotl/config/ | |
| - ./outputs:/workspace/axolotl/outputs/ | |
| - ~/.cache/huggingface/:/root/.cache/huggingface/ | |
| # set environment variables | |
| environment: | |
| - WANDB_API_KEY=${WANDB_API_KEY} | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| # count: 1 | |
| capabilities: [gpu] | |