fffiloni commited on
Commit
e4d0aa5
1 Parent(s): 618d462

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -0
inference.py CHANGED
@@ -122,6 +122,8 @@ if __name__ == "__main__":
122
 
123
  if args.is_long_video:
124
  window_size = int(np.sqrt(args.video_length))
 
 
125
  sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
126
  num_inference_steps=args.inference_steps, smooth_steps=args.smoother_steps, window_size=window_size,
127
  generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
 
122
 
123
  if args.is_long_video:
124
  window_size = int(np.sqrt(args.video_length))
125
+ if window_size > 4 :
126
+ window_size = 4
127
  sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
128
  num_inference_steps=args.inference_steps, smooth_steps=args.smoother_steps, window_size=window_size,
129
  generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,