compose.base.yml 660 B

12345678910111213141516171819202122232425
  1. services:
  2. app-base:
  3. build:
  4. context: .
  5. dockerfile: docker/Dockerfile
  6. args:
  7. BACKEND: ${BACKEND:-cuda} # or cpu
  8. CUDA_VER: ${CUDA_VER:-12.9.0}
  9. UV_EXTRA: ${UV_EXTRA:-cu129}
  10. UV_VERSION: ${UV_VERSION:-0.8.15}
  11. volumes:
  12. - /root/fish-checkpoints:/app/checkpoints
  13. - /root/fish-references/s2-pro:/app/references
  14. environment:
  15. COMPILE: ${COMPILE:-0}
  16. # GPU (remove this block if CPU-only):
  17. deploy:
  18. resources:
  19. reservations:
  20. devices:
  21. - driver: nvidia
  22. count: all
  23. capabilities: [ gpu ]
  24. tty: true
  25. stdin_open: true