|
from videosys import LatteConfig, VideoSysEngine |
|
|
|
|
|
def run_base(): |
|
config = LatteConfig(world_size=1) |
|
engine = VideoSysEngine(config) |
|
|
|
prompt = "Sunset over the sea." |
|
video = engine.generate(prompt).video[0] |
|
engine.save_video(video, f"./outputs/{prompt}.mp4") |
|
|
|
|
|
def run_pab(): |
|
config = LatteConfig(world_size=1) |
|
engine = VideoSysEngine(config) |
|
|
|
prompt = "Sunset over the sea." |
|
video = engine.generate(prompt).video[0] |
|
engine.save_video(video, f"./outputs/{prompt}.mp4") |
|
|
|
|
|
if __name__ == "__main__": |
|
run_base() |
|
|
|
|
