+9
7bc1dae095
7bc1dae095
Co-authored-by: yhyang201 <yhyang201@gmail.com> Co-authored-by: yizhang2077 <1109276519@qq.com> Co-authored-by: Xinyuan Tong <xinyuantong.cs@gmail.com> Co-authored-by: ispobock <ispobaoke@gmail.com> Co-authored-by: JiLi <leege233@gmail.com> Co-authored-by: CHEN Xi <78632976+RubiaCx@users.noreply.github.com> Co-authored-by: laixin <xielx@shanghaitech.edu.cn> Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com> Co-authored-by: jzhang38 <a1286225768@gmail.com> Co-authored-by: BrianChen1129 <yongqichcd@gmail.com> Co-authored-by: Kevin Lin <42618777+kevin314@users.noreply.github.com> Co-authored-by: Edenzzzz <wtan45@wisc.edu> Co-authored-by: rlsu9 <r3su@ucsd.edu> Co-authored-by: Jinzhe Pan <48981407+eigensystem@users.noreply.github.com> Co-authored-by: foreverpiano <pianoqwz@qq.com> Co-authored-by: RandNMR73 <notomatthew31@gmail.com> Co-authored-by: PorridgeSwim <yz3883@columbia.edu> Co-authored-by: Jiali Chen <90408393+gary-chenjl@users.noreply.github.com>
38 lines
949 B
Python
38 lines
949 B
Python
# Copied and adapted from: https://github.com/hao-ai-lab/FastVideo
|
|
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
from dataclasses import dataclass, field
|
|
|
|
from sglang.multimodal_gen.configs.sample.base import SamplingParams
|
|
from sglang.multimodal_gen.configs.sample.teacache import TeaCacheParams
|
|
|
|
|
|
@dataclass
|
|
class HunyuanSamplingParams(SamplingParams):
|
|
num_inference_steps: int = 50
|
|
|
|
num_frames: int = 125
|
|
height: int = 720
|
|
width: int = 1280
|
|
fps: int = 24
|
|
|
|
guidance_scale: float = 1.0
|
|
|
|
teacache_params: TeaCacheParams = field(
|
|
default_factory=lambda: TeaCacheParams(
|
|
teacache_thresh=0.15,
|
|
coefficients=[
|
|
7.33226126e02,
|
|
-4.01131952e02,
|
|
6.75869174e01,
|
|
-3.14987800e00,
|
|
9.61237896e-02,
|
|
],
|
|
)
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class FastHunyuanSamplingParam(HunyuanSamplingParams):
|
|
num_inference_steps: int = 6
|