16 lines
527 B
Python
16 lines
527 B
Python
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}")
|