"480 is more than the time allowed": we have to reduce the inference steps (#4)
Browse files- "480 is more than the time allowed": we have to reduce the inference steps (cafbe8aec358c469c14bb0bc64eb6b6f58bd4497)
Co-authored-by: Fabrice TIERCELIN <[email protected]>
app.py
CHANGED
|
@@ -19,7 +19,7 @@ interface = gr.Interface(
|
|
| 19 |
theme="soft"
|
| 20 |
)
|
| 21 |
|
| 22 |
-
@spaces.GPU(duration=
|
| 23 |
def generate_video(image):
|
| 24 |
"""
|
| 25 |
Generates a video from an input image using the pipeline.
|
|
@@ -30,7 +30,7 @@ def generate_video(image):
|
|
| 30 |
Returns:
|
| 31 |
The path of a video file.
|
| 32 |
"""
|
| 33 |
-
video_frames = pipeline(image=image, num_inference_steps=
|
| 34 |
|
| 35 |
# Frames to Video
|
| 36 |
os.makedirs("outputs", exist_ok=True)
|
|
|
|
| 19 |
theme="soft"
|
| 20 |
)
|
| 21 |
|
| 22 |
+
@spaces.GPU(duration=360)
|
| 23 |
def generate_video(image):
|
| 24 |
"""
|
| 25 |
Generates a video from an input image using the pipeline.
|
|
|
|
| 30 |
Returns:
|
| 31 |
The path of a video file.
|
| 32 |
"""
|
| 33 |
+
video_frames = pipeline(image=image, num_inference_steps=10).images
|
| 34 |
|
| 35 |
# Frames to Video
|
| 36 |
os.makedirs("outputs", exist_ok=True)
|