From c8cb932dc97bccad5b2184c5f32365c12017685c Mon Sep 17 00:00:00 2001
From: Liu Feng <42316773+LiewFeng@users.noreply.github.com>
Date: Wed, 18 Dec 2024 10:49:43 +0800
Subject: [PATCH] Update utils.py

---
 eval/teacache/experiments/utils.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/eval/teacache/experiments/utils.py b/eval/teacache/experiments/utils.py
index cb52309..01e3813 100644
--- a/eval/teacache/experiments/utils.py
+++ b/eval/teacache/experiments/utils.py
@@ -10,8 +10,7 @@ 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]
+            video = pipeline.generate(prompt, seed=l, **kwargs).video[0]
             pipeline.save_video(video, os.path.join(output_dir, f"{prompt}-{l}.mp4"))