Add isolated HeartMuLa 3B quality test
This commit is contained in:
1 parent
7c95dab324
commit
fe2a93eeb1
5 files changed
+253
-1
No files matched your search
@@ -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
|
||||
Reference in new issue
Block a user