fffiloni commited on
Commit
1def84e
·
1 Parent(s): 8d9eff1

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -1
inference.py CHANGED
@@ -118,7 +118,8 @@ if __name__ == "__main__":
118
  # Step 3. inference
119
 
120
  if args.is_long_video:
121
- window_size = int(np.sqrt(args.video_length))
 
122
  sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
123
  num_inference_steps=50, smooth_steps=args.smoother_steps, window_size=window_size,
124
  generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
 
118
  # Step 3. inference
119
 
120
  if args.is_long_video:
121
+ import math
122
+ window_size = int(math.sqrt(args.video_length))
123
  sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
124
  num_inference_steps=50, smooth_steps=args.smoother_steps, window_size=window_size,
125
  generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,