from .enhance import enhance_score
from .globals import (
    enable_enhance,
    get_enhance_weight,
    get_num_frames,
    is_enhance_enabled,
    set_enhance_weight,
    set_num_frames,
)
from .models.cogvideox import inject_enhance_for_cogvideox
from .models.hunyuanvideo import inject_enhance_for_hunyuanvideo

__all__ = [
    "inject_enhance_for_cogvideox",
    "inject_enhance_for_hunyuanvideo",
    "enhance_score",
    "get_num_frames",
    "set_num_frames",
    "get_enhance_weight",
    "set_enhance_weight",
    "enable_enhance",
    "is_enhance_enabled",
]