Integrate pinned YuE2 community WebUI
This commit is contained in:
1 parent
72c9d8c485
commit
af25425aee
4 files changed
+165
-17
No files matched your search
@@ -5,6 +5,7 @@ services:
|
||||
context: .
|
||||
args:
|
||||
YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
|
||||
YUE2_WEBUI_COMMIT: 8fc05609bde5dcd345d7c6d57fff3da0839164c5
|
||||
image: mike-ai/yue2:3b-0.1.6
|
||||
restart: "no"
|
||||
environment:
|
||||
@@ -23,22 +24,63 @@ services:
|
||||
capabilities: [gpu]
|
||||
|
||||
yue2-ui:
|
||||
profiles: ["playground"]
|
||||
profiles: ["community-ui"]
|
||||
build:
|
||||
context: .
|
||||
args:
|
||||
YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
|
||||
YUE2_WEBUI_COMMIT: 8fc05609bde5dcd345d7c6d57fff3da0839164c5
|
||||
image: mike-ai/yue2:3b-0.1.6
|
||||
container_name: mike-ai-yue2-playground
|
||||
restart: "no"
|
||||
entrypoint: ["python", "/opt/yue2/webui/server.py"]
|
||||
environment:
|
||||
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
|
||||
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
||||
YUE2_HOST: "0.0.0.0"
|
||||
YUE2_PORT: "8014"
|
||||
YUE2_OUTPUTS: "/workspace/runs"
|
||||
HF_HUB_OFFLINE: "1"
|
||||
TRANSFORMERS_OFFLINE: "1"
|
||||
ports:
|
||||
- "127.0.0.1:8014:8014"
|
||||
volumes:
|
||||
- /data/models/yue2:/opt/yue2/models:ro
|
||||
- /data/music/yue2:/workspace/runs
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
|
||||
capabilities: [gpu]
|
||||
healthcheck:
|
||||
test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8014/api/state', timeout=2)"]
|
||||
interval: 5s
|
||||
timeout: 3s
|
||||
retries: 12
|
||||
start_period: 5s
|
||||
|
||||
# Kept as a deliberately non-default fallback until the community console
|
||||
# has completed a real generation on Athena.
|
||||
yue2-playground-fallback:
|
||||
profiles: ["playground-fallback"]
|
||||
build:
|
||||
context: .
|
||||
args:
|
||||
YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
|
||||
YUE2_WEBUI_COMMIT: 8fc05609bde5dcd345d7c6d57fff3da0839164c5
|
||||
image: mike-ai/yue2:3b-0.1.6
|
||||
container_name: mike-ai-yue2-playground-fallback
|
||||
restart: "no"
|
||||
entrypoint: ["python", "/opt/yue2-playground/server.py"]
|
||||
environment:
|
||||
NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
|
||||
NVIDIA_DRIVER_CAPABILITIES: compute,utility
|
||||
YUE2_UI_HOST: "0.0.0.0"
|
||||
YUE2_UI_PORT: "8014"
|
||||
YUE2_UI_PORT: "8016"
|
||||
ports:
|
||||
- "127.0.0.1:8014:8014"
|
||||
- "127.0.0.1:8016:8016"
|
||||
volumes:
|
||||
- /data/models/yue2:/workspace/models:ro
|
||||
- /data/music/yue2:/workspace/runs
|
||||
@@ -49,9 +91,3 @@ services:
|
||||
- driver: nvidia
|
||||
device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
|
||||
capabilities: [gpu]
|
||||
healthcheck:
|
||||
test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8014/health', timeout=2)"]
|
||||
interval: 5s
|
||||
timeout: 3s
|
||||
retries: 12
|
||||
start_period: 5s
|
||||
Reference in new issue
Block a user