jjz5463 commited on
Commit
37a978c
·
1 Parent(s): 45f2ffe

lower inference steps

Browse files
Files changed (1) hide show
  1. 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=50,
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),