docker-compose.yml

services:
  dev:
    build:
      context: .
      dockerfile: Dockerfile.dev
    image: mini-transformer:dev
    container_name: mini_transformer_dev
    working_dir: /app
    volumes:
     - ./outputs:/app/outputs
     - ./trained_models:/app/trained_models
    environment:
      MINI_TRANSFORMER_MODELS: /app/trained_models
      MINI_TRANSFORMER_UI_HOST: 0.0.0.0
      MINI_TRANSFORMER_UI_PORT: 8000
    stdin_open: true
    tty: true
    gpus: all
    command: bash
    ports:
      - "8000:8000"

  infer:
    build:
      context: .
      dockerfile: Dockerfile.infer
    image: mini-transformer:infer
    container_name: mini_transformer_infer
    working_dir: /app
    volumes:
     - ./outputs:/app/outputs
     - ./trained_models:/app/trained_models
    environment:
      MINI_TRANSFORMER_MODELS: /app/trained_models
      MINI_TRANSFORMER_UI_HOST: 0.0.0.0
      MINI_TRANSFORMER_UI_PORT: 8000
    stdin_open: true
    tty: true
    ports:
      - "8000:8000"