Upload docker-compose.yml with huggingface_hub
Browse files- docker-compose.yml +25 -0
docker-compose.yml
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version: "3.8"
|
| 2 |
+
|
| 3 |
+
services:
|
| 4 |
+
ai-toolkit:
|
| 5 |
+
image: ostris/aitoolkit:latest
|
| 6 |
+
restart: unless-stopped
|
| 7 |
+
ports:
|
| 8 |
+
- "8675:8675"
|
| 9 |
+
volumes:
|
| 10 |
+
- ~/.cache/huggingface/hub:/root/.cache/huggingface/hub
|
| 11 |
+
- ./aitk_db.db:/app/ai-toolkit/aitk_db.db
|
| 12 |
+
- ./datasets:/app/ai-toolkit/datasets
|
| 13 |
+
- ./output:/app/ai-toolkit/output
|
| 14 |
+
- ./config:/app/ai-toolkit/config
|
| 15 |
+
environment:
|
| 16 |
+
- AI_TOOLKIT_AUTH=${AI_TOOLKIT_AUTH:-password}
|
| 17 |
+
- NODE_ENV=production
|
| 18 |
+
- TZ=UTC
|
| 19 |
+
deploy:
|
| 20 |
+
resources:
|
| 21 |
+
reservations:
|
| 22 |
+
devices:
|
| 23 |
+
- driver: nvidia
|
| 24 |
+
count: all
|
| 25 |
+
capabilities: [gpu]
|