LPX55 commited on
Commit
f16c010
·
verified ·
1 Parent(s): 1953cee

Update app_v2v.py

Browse files
Files changed (1) hide show
  1. app_v2v.py +1 -2
app_v2v.py CHANGED
@@ -279,7 +279,6 @@ def set_mp4_comments_imageio_ffmpeg(input_file, comments):
279
  print(f"Error saving prompt to video metadata, ffmpeg may be required: "+str(e))
280
  return False
281
 
282
- @spaces.GPU()
283
  @torch.no_grad()
284
  def worker(input_video, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
285
 
@@ -531,7 +530,7 @@ def worker(input_video, prompt, n_prompt, seed, batch, resolution, total_second_
531
  stream.output_queue.push(('end', None))
532
  return
533
 
534
- @spaces.GPU()
535
  def process(input_video, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
536
  global stream, high_vram
537
  # 20250506 pftq: Updated assertion for video input
 
279
  print(f"Error saving prompt to video metadata, ffmpeg may be required: "+str(e))
280
  return False
281
 
 
282
  @torch.no_grad()
283
  def worker(input_video, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
284
 
 
530
  stream.output_queue.push(('end', None))
531
  return
532
 
533
+ @spaces.GPU(duration=90)
534
  def process(input_video, prompt, n_prompt, seed, batch, resolution, total_second_length, latent_window_size, steps, cfg, gs, rs, gpu_memory_preservation, use_teacache, no_resize, mp4_crf, num_clean_frames, vae_batch):
535
  global stream, high_vram
536
  # 20250506 pftq: Updated assertion for video input