Maximofn commited on
Commit
7cd9249
·
verified ·
1 Parent(s): f162bb2

Update app.py

Browse files

DOwnl inference zero GPU time

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def save_video(video_frames, fps=25):
65
  export_to_video(video_frames, tmp_file.name, fps=fps)
66
  return tmp_file.name
67
 
68
- @spaces.GPU(duration=1500)
69
  def generate_video_pipeline(pipeline, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
70
  """Generate video using the pipeline"""
71
  print(f"Is CUDA available: {torch.cuda.is_available()}")
 
65
  export_to_video(video_frames, tmp_file.name, fps=fps)
66
  return tmp_file.name
67
 
68
+ @spaces.GPU(duration=600)
69
  def generate_video_pipeline(pipeline, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
70
  """Generate video using the pipeline"""
71
  print(f"Is CUDA available: {torch.cuda.is_available()}")