ezlocalai/docker-compose-rocm.yml at main · DevXT-LLC/ezlocalai · 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
services:
  ezlocalai:
    container_name: ezlocalai
    build:
      context: .
      dockerfile: rocm.Dockerfile
    env_file:
      - .env
    environment:
      - EZLOCALAI_URL=${EZLOCALAI_URL:-http://localhost:8091}
      - EZLOCALAI_API_KEY=${EZLOCALAI_API_KEY:-}
      - DEFAULT_MODEL=${DEFAULT_MODEL:-unsloth/Qwen3.5-4B-GGUF}
      - WHISPER_MODEL=${WHISPER_MODEL:-large-v3}
      - IMG_MODEL=${IMG_MODEL:-}
      - VIDEO_MODEL=${VIDEO_MODEL:-}
      - LLM_BATCH_SIZE=${LLM_BATCH_SIZE:-2048}
      - TOKENIZERS_PARALLELISM=false
      - MAX_CONCURRENT_REQUESTS=${MAX_CONCURRENT_REQUESTS:-1}
      - MAX_QUEUE_SIZE=${MAX_QUEUE_SIZE:-100}
      - REQUEST_TIMEOUT=${REQUEST_TIMEOUT:-300}
      - VOICE_SERVER=${VOICE_SERVER:-}
      - VOICE_SERVER_API_KEY=${VOICE_SERVER_API_KEY:-}
      - IMAGE_SERVER=${IMAGE_SERVER:-}
      - IMAGE_SERVER_API_KEY=${IMAGE_SERVER_API_KEY:-}
      - TEXT_SERVER=${TEXT_SERVER:-}
      - TEXT_SERVER_API_KEY=${TEXT_SERVER_API_KEY:-}
      - LAZY_LOAD_VOICE=${LAZY_LOAD_VOICE:-true}
      # ROCm specific environment variables
      - HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION:-11.0.0}
      - ROCM_PATH=/opt/rocm
      - HIP_VISIBLE_DEVICES=${HIP_VISIBLE_DEVICES:-0}
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-sf", "http://localhost:8091/health"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 120s
    logging:
      driver: json-file
      options:
        max-size: "500m"
        max-file: "5"
    ulimits:
      nofile:
        soft: 65535
        hard: 65535
    ports:
      - "8091:8091"
    volumes:
      - ./models:/app/models
      - ./hf:/root/.cache/huggingface/hub
      - ./outputs:/app/outputs
      - ./voices:/app/voices
      - ./whispercpp:/app/whispercpp
    devices:
      - /dev/kfd:/dev/kfd
      - /dev/dri:/dev/dri
    group_add:
      - video
      - render
    security_opt:
      - seccomp=unconfined
    cap_add:
      - SYS_PTRACE