dream

1
name: dialogrpt-ru
2
endpoints:
3
- respond
4
compose:
5
  env_file:
6
  - .env_ru
7
  build:
8
    context: ./services/dialogrpt_ru/
9
    args:
10
      SERVICE_PORT: 8122
11
      PRETRAINED_MODEL_FNAME: dialogrpt_ru_ckpt_v0.pth
12
      TOKENIZER_NAME_OR_PATH: DeepPavlov/rudialogpt3_medium_based_on_gpt2_v2
13
      CUDA_VISIBLE_DEVICES: '0'
14
      FLASK_APP: server
15
  command: flask run -h 0.0.0.0 -p 8122
16
  environment:
17
  - CUDA_VISIBLE_DEVICES=0
18
  - FLASK_APP=server
19
  deploy:
20
    resources:
21
      limits:
22
        memory: 4G
23
      reservations:
24
        memory: 4G
25
  volumes:
26
  - ./services/dialogrpt_ru:/src
27
  - ~/.deeppavlov/cache:/root/.cache
28
  ports:
29
  - 8122:8122
30
proxy: null
31