update

docker-compose.yml

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