File size: 973 Bytes
dc8d52b
 
 
 
 
2d52135
 
dc8d52b
 
2d52135
 
 
 
 
 
dc8d52b
 
 
 
2d52135
 
b22e8ba
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
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"