services:
  producer:
    image: producer:latest
    network_mode: "host"
    volumes:
      - /dockerdata/task/upload:/obscura/task/upload
      - /dockerdata/task/result:/obscura/task/result
      - /dockerdata/producer/config.py:/obscura/producer/config.py
    environment:
      - TZ=Asia/Shanghai
      - CUDA_VISIBLE_DEVICES=2,3
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

  producer-chat:
    image: producer_chat:latest
    network_mode: "host"
    volumes:
      - /dockerdata/task/upload:/obscura/task/upload
      - /dockerdata/producer_chat/.env:/obscura/producer_chat/.env
      - /dockerdata/task/audio_files:/obscura/task/audio_files
    environment:
      - TZ=Asia/Shanghai
      - CUDA_VISIBLE_DEVICES=2,3
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]