stable-diffusion-docker/docker-compose.yml at main · Luxbit/stable-diffusion-docker · 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
version: "3.4"

services:
  stable-diffusion:
    build:
      context: ./
      dockerfile: Dockerfile
    image: ghcr.io/luxbit/stable-diffusion-docker:latest
    ports:
      - 5000:5000
    container_name: stable-diffusion
    environment:
      PROMPT: "A high tech solarpunk utopia in the Amazon rainforest"
      N_SAMPLES: 1
      HEIGHT: 512
      WIDTH: 512
      SCALE: 7.5
      DDIM_STEPS: 50
      SEED: 0
      SAFETY_CHECK: 'True'
      FLOAT32: 'True'
    volumes:
      - ./huggingface:/home/huggingface/.cache/huggingface
      - ./output:/home/huggingface/output
    command: bash
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]