xinyun/TeaCache
1
0
Fork 0
mirror of https://git.datalinker.icu/ali-vilab/TeaCache synced 2026-05-02 22:11:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
Liu Feng 2024-12-18 10:49:43 +08:00 committed by GitHub
parent 21ea48e71a
commit c8cb932dc9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
eval/teacache/experiments/utils.py
View File

@ -10,8 +10,7 @@ def generate_func(pipeline, prompt_list, output_dir, loop: int = 5, kwargs: dict
kwargs["verbose"] = False kwargs["verbose"] = False
for prompt in tqdm.tqdm(prompt_list): for prompt in tqdm.tqdm(prompt_list):
for l in range(loop): for l in range(loop):
set_seed(l) video = pipeline.generate(prompt, seed=l, **kwargs).video[0]
video = pipeline.generate(prompt, **kwargs).video[0]
pipeline.save_video(video, os.path.join(output_dir, f"{prompt}-{l}.mp4")) pipeline.save_video(video, os.path.join(output_dir, f"{prompt}-{l}.mp4"))