Prepare isolated YuE2 music evaluation

2026-09-10 21:25:59 +02:00
parent 6bd6c48a95
commit 1904f2104f
9 changed files with 102 additions and 263 deletions
@@ -0,0 +1,23 @@
+services:
+  yue2-test:
+    profiles: ["manual"]
+    build:
+      context: .
+      args:
+        YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
+    image: mike-ai/yue2:3b-0.1.6
+    restart: "no"
+    environment:
+      NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
+      NVIDIA_DRIVER_CAPABILITIES: compute,utility
+    volumes:
+      - /data/models/yue2:/workspace/models:ro
+      - /data/music/yue2:/workspace/runs
+      - ./requests:/workspace/requests:ro
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
+              capabilities: [gpu]