diff --git a/docker-compose.yml b/docker-compose.yml
index 07c32d4..1a21b5c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -10,4 +10,11 @@ services:
       - LOGGER_LEVEL=debug
       - LISTEN_IP=0.0.0.0
       - LISTEN_PORT=8000
-      - REPO_DIR=/home/app/repo/
\ No newline at end of file
+      - REPO_DIR=/home/app/repo/
+    # deploy:
+    #   resources:
+    #     reservations:
+    #       devices:
+    #         - driver: nvidia
+    #           device_ids: ['0']
+    #           capabilities: [gpu]
\ No newline at end of file