Add isolated HeartMuLa 3B quality test

This commit is contained in:
Mikei386 committed 2026-09-10 20:20:51 +02:00
1 parent 7c95dab324
commit fe2a93eeb1
5 files changed
+253 -1

No files matched your search

+32
View File
@@ -0,0 +1,32 @@
services:
heartmula-test:
build:
context: .
args:
HEARTLIB_COMMIT: 3783bdb8441f2c298b1e64c8651173aac200361c
image: mike-ai/heartmula:3b-3783bdb
container_name: mike-ai-heartmula-test
restart: "no"
environment:
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe,GPU-4834d9d7-5b61-3004-1fb3-4ae49d482d4b"
NVIDIA_DRIVER_CAPABILITIES: compute,utility
HEARTMULA_MULA_DEVICE: cuda:0
HEARTMULA_CODEC_DEVICE: cuda:1
volumes:
- /data/models/heartmula:/models:ro
- /data/music/heartmula:/output
ports:
- "7863:7860"
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
healthcheck:
test: ["CMD", "curl", "-fsS", "http://127.0.0.1:7860/healthz"]
interval: 15s
timeout: 5s
retries: 20
start_period: 180s