yumoqing/pipeline-app
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: ktv_adapter model names — remove ky- prefix to match actual Sage llm records

Browse Source 
- All 8 GPU service models now match tokentest llm names
- Add supporting app modules (quality_gate, model_selector, global_func, eval_music, eval_video)
This commit is contained in:
yumoqing 2026-07-08 18:05:13 +08:00
parent bba2e12cbf
commit a243f1d9cf
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
app/ktv_adapter.py
View File

@ -6,14 +6,14 @@
KTV_FILE_BASE: 文件服务地址，默认同 KTV_API_BASE 所在域
步骤→模型映射（KTV_STEP_MODELS 字典，可通过环境变量覆盖各项）:
demucs_separating → ky-demucs-separate
asr_transcribing → ky-asr-transcribe
subtitle_rendering → ky-subtitle-render
face_detecting → ky-face-detect
realesrgan_upscaling → ky-realesrgan-upscale
rvc_converting → ky-rvc-convert
video_evaluating → ky-video-eval-evaluate
songrate_evaluating → ky-songrate-evaluate
demucs_separating → demucs-separate
asr_transcribing → asr-transcribe
subtitle_rendering → subtitle-render
face_detecting → face-detect
realesrgan_upscaling → realesrgan-upscale
rvc_converting → rvc-convert
video_evaluating → video-eval-evaluate
songrate_evaluating → songrate-evaluate
本地操作（ffmpeg/ffprobe）不走 API。
"""
@ -40,14 +40,14 @@ API_KEY = os.environ.get("KTV_API_KEY", "")
FILE_BASE = os.environ.get("KTV_FILE_BASE", API_BASE.rsplit("/", 2)[0])
STEP_MODELS = {
"demucs_separating": os.environ.get("KTV_MODEL_DEMUCS", "ky-demucs-separate"),
"asr_transcribing": os.environ.get("KTV_MODEL_ASR", "ky-asr-transcribe"),
"subtitle_rendering": os.environ.get("KTV_MODEL_SUBTITLE", "ky-subtitle-render"),
"face_detecting": os.environ.get("KTV_MODEL_FACE", "ky-face-detect"),
"realesrgan_upscaling": os.environ.get("KTV_MODEL_REALESRGAN","ky-realesrgan-upscale"),
"rvc_converting": os.environ.get("KTV_MODEL_RVC", "ky-rvc-convert"),
"video_evaluating": os.environ.get("KTV_MODEL_VIDEO_EVAL","ky-video-eval-evaluate"),
"songrate_evaluating": os.environ.get("KTV_MODEL_SONGRATE", "ky-songrate-evaluate"),
"demucs_separating": os.environ.get("KTV_MODEL_DEMUCS", "demucs-separate"),
"asr_transcribing": os.environ.get("KTV_MODEL_ASR", "asr-transcribe"),
"subtitle_rendering": os.environ.get("KTV_MODEL_SUBTITLE", "subtitle-render"),
"face_detecting": os.environ.get("KTV_MODEL_FACE", "face-detect"),
"realesrgan_upscaling": os.environ.get("KTV_MODEL_REALESRGAN","realesrgan-upscale"),
"rvc_converting": os.environ.get("KTV_MODEL_RVC", "rvc-convert"),
"video_evaluating": os.environ.get("KTV_MODEL_VIDEO_EVAL","video-eval-evaluate"),
"songrate_evaluating": os.environ.get("KTV_MODEL_SONGRATE", "songrate-evaluate"),
}
LOCAL_WORK_DIR = "/data/pipeline/ktv"
@ -197,7 +197,7 @@ async def handle_video_preparing(tenant_id, task_id, step_name, input_data, conf
# ── Demucs ─────────────────────────────────────────────────────────
async def handle_demucs_separating(tenant_id, task_id, step_name, input_data, config):
"""通过 API 调 ky-demucs-separate。固定使用 separate_full 模式。"""
"""通过 API 调 demucs-separate。固定使用 separate_full 模式。"""
work_dir = _task_dir(task_id)
audio_path = None
for dep_output in input_data.values():