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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
name: quadtrix

x-common-env: &common-env
  TZ: UTC
  PYTHONUNBUFFERED: "1"

services:

  frontend:
    build:
      context: .
      dockerfile: .devops/Dockerfile.frontend
      args:
        VITE_API_BASE_URL: ""
    image: quadtrix/frontend:latest
    container_name: quadtrix-frontend
    restart: unless-stopped
    ports:
      - "5173:80"
    depends_on:
      backend:
        condition: service_healthy
    networks:
      - quadtrix-net
    healthcheck:
      test: [ "CMD", "wget", "-qO-", "http://localhost/" ]
      interval: 30s
      timeout: 5s
      retries: 3

  backend:
    build:
      context: .
      dockerfile: .devops/Dockerfile.backend
    image: quadtrix/backend:latest
    container_name: quadtrix-backend
    restart: unless-stopped
    ports:
      - "3001:3001"
    environment:
      <<: *common-env
      API_PORT: "3001"
      CORS_ORIGINS: "http://localhost:5173,http://frontend"
      REDIS_URL: "redis://redis:6379/0"
      TORCH_CHECKPOINT_PATH: /models/best_model.pt
      LOG_LEVEL: INFO
      MAX_SESSIONS: "500"
      SESSION_TTL_HOURS: "24"
    volumes:
      - models:/models
      - ./engine:/app/engine:ro
    depends_on:
      redis:
        condition: service_healthy
    networks:
      - quadtrix-net
    healthcheck:
      test: [ "CMD", "curl", "-f", "http://localhost:3001/api/health" ]
      interval: 30s
      timeout: 10s
      start_period: 20s
      retries: 3

  redis:
    image: redis:7-alpine
    container_name: quadtrix-redis
    restart: unless-stopped
    command: redis-server --maxmemory 256mb --maxmemory-policy allkeys-lru
    volumes:
      - redis-data:/data
    networks:
      - quadtrix-net
    healthcheck:
      test: [ "CMD", "redis-cli", "ping" ]
      interval: 10s
      timeout: 5s
      retries: 5
    expose:
      - "6379"

  cpp:
    build:
      context: .
      dockerfile: .devops/Dockerfile.cpp
    image: quadtrix/cpp:latest
    container_name: quadtrix-cpp

    restart: "no"
    stdin_open: true
    tty: true
    volumes:
      - models:/models
      - ./data:/app/data:ro
    environment:
      <<: *common-env
      GPT_DATA_PATH: /app/data/input.txt
      GPT_MODEL_PATH: /models/best_model.bin
    networks:
      - quadtrix-net
    profiles:
      - cpp

  train-cpp:
    build:
      context: .
      dockerfile: .devops/Dockerfile.cpp
    image: quadtrix/cpp:latest
    container_name: quadtrix-train-cpp
    restart: "no"
    volumes:
      - models:/models
      - ./data:/app/data:ro
    environment:
      <<: *common-env
      GPT_DATA_PATH: /app/data/input.txt
      GPT_MODEL_PATH: /models/best_model.bin
    command: [ "data/input.txt" ] # train mode (no --chat flag)
    networks:
      - quadtrix-net
    profiles:
      - train

  train-torch:
    build:
      context: .
      dockerfile: .devops/Dockerfile.backend
    image: quadtrix/backend:latest
    container_name: quadtrix-train-torch
    restart: "no"
    volumes:
      - models:/models
      - ./engine:/app/engine
      - ./data:/app/data:ro
    environment:
      <<: *common-env
      QUADTRIX_TRAIN_DATA: /app/data/input.txt
    working_dir: /app
    command: [ "python", "engine/main.py" ]
    networks:
      - quadtrix-net
    profiles:
      - train

  benchmark:
    build:
      context: .
      dockerfile: .devops/Dockerfile.cpp
    image: quadtrix/cpp:latest
    container_name: quadtrix-benchmark
    restart: "no"
    volumes:
      - models:/models
      - ./data:/app/data:ro
      - ./benchmark_results.csv:/app/benchmark_results.csv
    environment:
      <<: *common-env
      GPT_MODEL_PATH: /models/best_model.bin

    command: [ "data/input.txt", "--generate" ]
    networks:
      - quadtrix-net
    profiles:
      - benchmark

volumes:
  models:
    driver: local
  redis-data:
    driver: local

networks:
  quadtrix-net:
    driver: bridge