feat: 新增代码

This commit is contained in:
Daniel
2026-04-07 00:37:39 +08:00
commit 8d0b729f2f
29 changed files with 1768 additions and 0 deletions

View File

@@ -0,0 +1,15 @@
from app.backends.hunyuan_backend import HunyuanBackend
from app.backends.ltx_backend import LTXBackend
class ModelRouter:
def __init__(self, ltx_backend: LTXBackend, hunyuan_backend: HunyuanBackend):
self._ltx = ltx_backend
self._hunyuan = hunyuan_backend
def route(self, quality_mode: str):
if quality_mode == "preview":
return self._ltx
if quality_mode == "refine":
return self._hunyuan
raise ValueError(f"Unsupported quality_mode: {quality_mode}")