services:
  vincie:
    image: img2img:edit
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    ports:
      - "7860:7860"
    environment:
      GRADIO_SERVER_PORT: "7860"
      HF_HUB_CACHE: "/data/.cache/huggingface/hub"
      CKPT_ROOT: "/data/ckpt/VINCIE-3B"
      VINCIE_ROOT: "/data/VINCIE"
    volumes:
      - vincie_hub:/data/.cache/huggingface/hub
      - vincie_ckpt:/data/ckpt/VINCIE-3B
      - vincie_out:/app/outputs
      - vincie_repo:/data/VINCIE
volumes:
  vincie_hub: {}
  vincie_ckpt: {}
  vincie_out: {}
  vincie_repo: {}