docker-compose.yaml

services:
  loquito-mlflow:
    build:
      context: .
      dockerfile: mlflow.Dockerfile
    container_name: loquito-mlflow
    ports:
      - "5000:5000" # Map MLflow server port
    volumes:
      - ${LOQUITO_DATA_DIR:?LOQUITO_DATA_DIR must be set}/mlflow:/mlflow-data # Mount host directory as volume
    environment:
      - MLFLOW_BACKEND_STORE_URI=/mlflow-data
      - MLFLOW_DEFAULT_ARTIFACT_ROOT=/mlflow-data

  loquito-train:
    build:
      context: .
      dockerfile: train.Dockerfile
    shm_size: 128g
    volumes:
      - ${LOQUITO_DATA_DIR:?LOQUITO_DATA_DIR must be set}/lmdrive:/workspace/data/lmdrive # Mount for lmdrive dataset
      - ${LOQUITO_DATA_DIR:?LOQUITO_DATA_DIR must be set}/models:/workspace/data/models   # Mount for models
      - ../:/workspace/projects/loquito # Mount for project directory
    stdin_open: true
    tty: true
    deploy:
      resources:
          reservations:
            devices:
              - driver: nvidia
                count: all
                capabilities: [gpu]