| services: | |
| wcaguar: | |
| build: . | |
| gpus: all | |
| ports: | |
| - "7860:7860" | |
| volumes: | |
| - ./data:/data | |
| - ./outputs:/app/outputs | |
| - ./audit:/app/audit | |
| environment: | |
| - HF_HOME=/data/hf | |
| - TRANSFORMERS_CACHE=/data/hf | |
| - HUGGINGFACE_HUB_CACHE=/data/hf | |
| - HF_TOKEN=${HF_TOKEN} | |
| - WCAGUAR_AUDIT_DIR=/app/audit | |
| - WCAGUAR_OUTPUT_DIR=/app/outputs | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: all | |
| capabilities: [gpu] | |