File size: 654 Bytes
07c6a04 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import json
import os
import tqdm
from videosys.utils.utils import set_seed
def generate_func(pipeline, prompt_list, output_dir, loop: int = 5, kwargs: dict = {}):
kwargs["verbose"] = False
for prompt in tqdm.tqdm(prompt_list):
for l in range(loop):
set_seed(l)
video = pipeline.generate(prompt, **kwargs).video[0]
pipeline.save_video(video, os.path.join(output_dir, f"{prompt}-{l}.mp4"))
def read_prompt_list(prompt_list_path):
with open(prompt_list_path, "r") as f:
prompt_list = json.load(f)
prompt_list = [prompt["prompt_en"] for prompt in prompt_list]
return prompt_list
|