docker-compose.yml

version: "3.3"

services:
  traefik:
    image: "traefik:v2.4"
    container_name: "traefik"
    command:
      - "--api.insecure=true"
      - "--providers.docker=true"
      - "--providers.docker.exposedbydefault=false"
      - "--entrypoints.web.address=:80"
      - "--entrypoints.websecure.address=:443"
      - "--entrypoints.web.http.redirections.entryPoint.to=websecure"
      - "--entrypoints.web.http.redirections.entryPoint.scheme=https"
      - "--entrypoints.web.http.redirections.entrypoint.permanent=true"
      - "--certificatesresolvers.myresolver.acme.email=${LE_EMAIL}"
      - "--certificatesresolvers.myresolver.acme.storage=acme.json"
      - "--certificatesresolvers.myresolver.acme.httpchallenge.entrypoint=web"
    ports:
      - "80:80"
      - "443:443"
      - "8080:8080"
    volumes:
      - "/var/run/docker.sock:/var/run/docker.sock:ro"
      - "./acme.json:/acme.json"

  gpt-neo-api:
    build:
      context: ./api
      args:
        - GPT_MODEL=${GPT_MODEL}
    container_name: "gpt-neo-api"
    environment:
      - GPT_MODEL
    ipc: "host"
    runtime: nvidia # remove/comment line for CPU-only inference
    labels:
      - "traefik.enable=true"
      - "traefik.http.routers.gptneoapi.rule=Host(`${API_DOMAIN}`)"
      - "traefik.http.routers.gptneoapi.tls=true"
      - "traefik.http.routers.gptneoapi.tls.certresolver=myresolver"

  gpt-neo-ui:
    build:
      context: ./ui
    container_name: "gpt-neo-ui"
    environment:
      - API_DOMAIN
    labels:
      - "traefik.enable=true"
      - "traefik.http.routers.gptneoui.rule=Host(`${UI_DOMAIN}`)"
      - "traefik.http.routers.gptneoui.tls=true"
      - "traefik.http.routers.gptneoui.tls.certresolver=myresolver"