Spaces:
Paused
Paused
Update inference.py
Browse files- inference.py +2 -2
inference.py
CHANGED
|
@@ -121,8 +121,8 @@ if __name__ == "__main__":
|
|
| 121 |
# Step 3. inference
|
| 122 |
|
| 123 |
if args.is_long_video:
|
| 124 |
-
window_size = int(np.sqrt(args.video_length))
|
| 125 |
-
|
| 126 |
sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
|
| 127 |
num_inference_steps=args.inference_steps, smooth_steps=args.smoother_steps, window_size=window_size,
|
| 128 |
generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
|
|
|
|
| 121 |
# Step 3. inference
|
| 122 |
|
| 123 |
if args.is_long_video:
|
| 124 |
+
#window_size = int(np.sqrt(args.video_length))
|
| 125 |
+
window_size = int(3)
|
| 126 |
sample = pipe.generate_long_video(args.prompt + POS_PROMPT, video_length=args.video_length, frames=pil_annotation,
|
| 127 |
num_inference_steps=args.inference_steps, smooth_steps=args.smoother_steps, window_size=window_size,
|
| 128 |
generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
|