fffiloni commited on
Commit
0a986cd
·
1 Parent(s): d5a8d3d

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -122,7 +122,7 @@ if __name__ == "__main__":
122
  if args.is_long_video:
123
  window_size = int(np.sqrt(args.video_length))
124
  sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
125
- num_inference_steps=50, smooth_steps=args.smoother_steps, window_size=window_size,
126
  generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
127
  width=args.width, height=args.height
128
  ).videos
 
122
  if args.is_long_video:
123
  window_size = int(np.sqrt(args.video_length))
124
  sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
125
+ num_inference_steps=25, smooth_steps=args.smoother_steps, window_size=window_size,
126
  generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
127
  width=args.width, height=args.height
128
  ).videos