Quadtrix.cpp/docker-compose.yml at master · Eamon2009/Quadtrix.cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
services:
  quadtrix:
    image: ghcr.io/eamon2009/quadtrix.cpp:latest
    build:
      context: .
      dockerfile: Dockerfile
      args:
        # for cuda
        # BASE_IMAGE: nvidia/cuda:12.4.1-cudnn-runtime-ubuntu24.04
        BASE_IMAGE: ubuntu:24.04

    ports:
      - "3001:3001" # FastAPI backend
      - "8080:8080" # React frontend

    volumes:
      # Place best_model.pt and/or best_model.bin inside ./models/
      - ./models:/app/models

    environment:
      TORCH_CHECKPOINT_PATH: /app/models/best_model.pt
      GPT_MODEL_PATH: /app/models/best_model.bin
      CORS_ORIGINS: http://localhost:8080
      LOG_LEVEL: INFO
      MAX_SESSIONS: 1000
      SESSION_TTL_HOURS: 24
    restart: unless-stopped

    healthcheck:
      test: [ "CMD", "curl", "-f", "http://localhost:3001/api/health" ]
      interval: 30s
      timeout: 10s
      retries: 5
      start_period: 20s