mirror of
https://git.datalinker.icu/ali-vilab/TeaCache
synced 2025-12-10 05:14:24 +08:00
22 lines
638 B
Python
22 lines
638 B
Python
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):
|
|
video = pipeline.generate(prompt, seed=l, **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
|