Mochi1 / demo_test.py
Haoxin Chen
add videocontrol code
0757c55
raw
history blame
423 Bytes
class Text2Video():
def __init__(self, result_dir='./tmp/') -> None:
pass
def get_prompt(self, input_text, steps=50, model_index=0, eta=1.0, cfg_scale=15.0, lora_scale=1.0):
return '01.mp4'
class VideoControl:
def __init__(self) -> None:
pass
def get_video(self, input_video, input_prompt, frame_stride, vc_steps, vc_cfg_scale, vc_eta):
return 'su','01.mp4'