File size: 571 Bytes
a6d4ba6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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",
]