From cc0e14e677443ccd4cacfb38f96f4ba1fe40134a Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Tue, 17 Jun 2025 00:22:36 +0800
Subject: [PATCH] update

---
 docker-compose.yml | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index fd5bf70a..b2563790 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,13 +11,13 @@ services:
       - ./src:/app/src
       - ./saves:/app/saves
       - ${MODEL_DIR:-./models}:/models  # 使用默认值处理未定义的环境变量
-    # deploy:
-    #   resources:
-    #     reservations:
-    #       devices:
-    #         - driver: nvidia
-    #           device_ids: ['1']
-    #           capabilities: [gpu]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: ['1']
+              capabilities: [gpu]
     ports:
       - "5050:5050"
     networks: