docker-compose.yml 447 B

123456789101112131415161718192021
  1. version: '3.9'
  2. services:
  3. model:
  4. build: .
  5. ports:
  6. - "7860:7860"
  7. volumes:
  8. - ../cache:/cache
  9. - ../output:/output
  10. - ../models:/models
  11. - ./config.json:/docker/config.json
  12. environment:
  13. - CLI_ARGS=--medvram --opt-split-attention
  14. deploy:
  15. resources:
  16. reservations:
  17. devices:
  18. - driver: nvidia
  19. device_ids: ['0']
  20. capabilities: [gpu]