From 0ede7fe071c149d1b646d9b82207b1b52e9d7bbe Mon Sep 17 00:00:00 2001 From: Dylan De Faoite Date: Tue, 3 Mar 2026 14:18:43 +0000 Subject: [PATCH] fix(compose): add GPU support to celery worker --- docker-compose.dev.yml | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml index 07a7085..04d435b 100644 --- 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: \ No newline at end of file