fix(compose): add GPU support to celery worker

2026-03-03 14:18:43 +00:00
parent eb4187c559
commit 0ede7fe071
1 changed files with 7 additions and 0 deletions
--- a/docker-compose.dev.yml
+++ b/docker-compose.dev.yml
@@ -48,6 +48,13 @@ services:
    depends_on:
      - postgres
      - redis
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]

 volumes:
  model_cache: