Update app.py
Browse filesDOwnl inference zero GPU time
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=
|
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()}")
|