examples/cogvideo/sample.py

from videosys import CogVideoConfig, VideoSysEngine


def run_base():
    config = CogVideoConfig(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()