diff --git a/docker-compose.yml b/docker-compose.yml
index 41a44fe..8b99705 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -3,12 +3,8 @@ services:
     image: xunlian:latest
     gpus: all
     volumes:
-      - ./yolov8n.pt:/workspace/yolov8n.pt:ro
-      - ./yolov8m.pt:/workspace/yolov8m.pt:ro
-      - ./yolo26n.pt:/workspace/yolo26n.pt:ro
+      - .:/workspace
       - /data/facepp:/data/facepp:ro
-
-      - ./runs:/workspace/runs
     environment:
       TZ: Asia/Shanghai
       NVIDIA_VISIBLE_DEVICES: all