Spaces:
Paused
Paused
lower inference steps
Browse files- baseline_utils.py +1 -1
baseline_utils.py
CHANGED
@@ -153,7 +153,7 @@ def generate_video(scene_list, writer_description, opt, fps=24): # Lower fps
|
|
153 |
video = pipe(
|
154 |
prompt=prompt + f'\nThe main character is described as: {writer_description}.',
|
155 |
num_videos_per_prompt=1,
|
156 |
-
num_inference_steps=
|
157 |
num_frames=fps,
|
158 |
guidance_scale=6,
|
159 |
generator=torch.Generator(device=device).manual_seed(42),
|
|
|
153 |
video = pipe(
|
154 |
prompt=prompt + f'\nThe main character is described as: {writer_description}.',
|
155 |
num_videos_per_prompt=1,
|
156 |
+
num_inference_steps=40,
|
157 |
num_frames=fps,
|
158 |
guidance_scale=6,
|
159 |
generator=torch.Generator(device=device).manual_seed(42),
|