Iceclear commited on
Commit
09fd3d7
·
verified ·
1 Parent(s): 273be34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -264,8 +264,8 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
264
  return positive_prompts_embeds
265
 
266
  def cut_videos(videos, sp_size):
267
- if videos.size(1) > 101:
268
- videos = videos[:, :101]
269
  t = videos.size(1)
270
  if t <= 4 * sp_size:
271
  print(f"Cut input video size: {videos.size()}")
@@ -443,7 +443,7 @@ with gr.Blocks(title="SeedVR2: One-Step Video Restoration via Diffusion Adversar
443
  </a>
444
 
445
  <h4>Notice</h4>
446
- <p>This demo supports up to <b>720p</b> and <b>101 frames</b>.
447
  For other use cases (image restoration, video resolutions beyond 720p, etc), check the <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>GitHub repo</a>.</p>
448
 
449
  <h4>Limitations</h4>
 
264
  return positive_prompts_embeds
265
 
266
  def cut_videos(videos, sp_size):
267
+ if videos.size(1) > 121:
268
+ videos = videos[:, :121]
269
  t = videos.size(1)
270
  if t <= 4 * sp_size:
271
  print(f"Cut input video size: {videos.size()}")
 
443
  </a>
444
 
445
  <h4>Notice</h4>
446
+ <p>This demo supports up to <b>720p</b> and <b>121 frames</b>.
447
  For other use cases (image restoration, video resolutions beyond 720p, etc), check the <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>GitHub repo</a>.</p>
448
 
449
  <h4>Limitations</h4>