WhisperLiveKit/compose.yml at main · QuentinFuxa/WhisperLiveKit · 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
services:
  wlk-gpu-sortformer:
    build:
      context: .
      dockerfile: Dockerfile
      args:
        EXTRAS: ${GPU_SORTFORMER_EXTRAS:-cu129,diarization-sortformer}
    image: wlk:gpu-sortformer
    gpus: all
    ports:
      - "8000:8000"
    volumes:
      - hf-cache:/root/.cache/huggingface/hub
      # - ${HF_TKN_FILE:-./token}:/root/.cache/huggingface/token:ro
    environment:
      - HF_TOKEN
    command: ["--model", "medium", "--diarization", "--pcm-input"]

  wlk-gpu-voxtral:
    build:
      context: .
      dockerfile: Dockerfile
      args:
        EXTRAS: ${GPU_VOXTRAL_EXTRAS:-cu129,voxtral-hf,translation}
    image: wlk:gpu-voxtral
    gpus: all
    ports:
      - "8001:8000"
    volumes:
      - hf-cache:/root/.cache/huggingface/hub
      # - ${HF_TKN_FILE:-./token}:/root/.cache/huggingface/token:ro
    environment:
      - HF_TOKEN
    command: ["--backend", "voxtral", "--pcm-input"]

  wlk-cpu:
    build:
      context: .
      dockerfile: Dockerfile.cpu
      args:
        EXTRAS: ${CPU_EXTRAS:-cpu,diarization-diart,translation}
    image: wlk:cpu
    ports:
      - "8000:8000"
    volumes:
      - hf-cache:/root/.cache/huggingface/hub
      # - ${HF_TKN_FILE:-./token}:/root/.cache/huggingface/token:ro
    environment:
      - HF_TOKEN

volumes:
  hf-cache: