version: '3.8'

services:
  jurifob-training:
    build: .
    container_name: jurifob-phd-training
    environment:
      - WANDB_API_KEY=${WANDB_API_KEY}
      - HF_TOKEN=${HF_TOKEN}
      - CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
    ports:
      - "7860:7860"
    volumes:
      - ./trabalho:/workspace/trabalho
      - ./checkpoints:/workspace/checkpoints
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 8
              capabilities: [gpu]
    shm_size: 64gb