multimodal-retrieval-lab/docker-compose.yml at main · KonNik88/multimodal-retrieval-lab · 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
services:
  qdrant-mm:
    image: qdrant/qdrant:latest
    container_name: qdrant-mm
    ports:
      - "6334:6333"
    volumes:
      - mm_qdrant_data:/qdrant/storage
    restart: unless-stopped

  mm-api:
    build:
      context: .
      dockerfile: docker/api/Dockerfile
    container_name: mm-api
    environment:
      QDRANT_URL: "http://qdrant-mm:6333"
      QDRANT_COLLECTION: "flickr8k_images"
      IMAGES_DIR: "/app/data/flickr8k/images"
      MODEL_NAME: "ViT-B-32"
      MODEL_PRETRAINED: "openai"
      DEVICE: "cpu"   # опционально: "cuda" (если настроен GPU passthrough в Docker)
      HF_HUB_DISABLE_SYMLINKS_WARNING: "1"
    ports:
      - "8008:8000"
    volumes:
      - ./data/flickr8k/images:/app/data/flickr8k/images:ro
    depends_on:
      - qdrant-mm
    restart: unless-stopped

  mm-ui:
    build:
      context: .
      dockerfile: docker/ui/Dockerfile
    container_name: mm-ui
    environment:
      API_URL: "http://mm-api:8000"
    ports:
      - "8508:8501"
    depends_on:
      - mm-api
    restart: unless-stopped

volumes:
  mm_qdrant_data: