Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
327e1e8
1
Parent(s):
0fa742d
out of memory! let's scale down to 3 seconds
Browse files
app.py
CHANGED
@@ -56,7 +56,8 @@ def infer(prompt, video_in, denoise_strength):
|
|
56 |
|
57 |
negative_prompt = "text, watermark, copyright, blurry, nsfw"
|
58 |
|
59 |
-
|
|
|
60 |
video_resized = [Image.fromarray(frame).resize((1024, 576)) for frame in video]
|
61 |
video_frames = pipe_xl(prompt, negative_prompt=negative_prompt, video=video_resized, strength=denoise_strength).frames
|
62 |
video_path = export_to_video(video_frames, output_video_path="xl_result.mp4")
|
|
|
56 |
|
57 |
negative_prompt = "text, watermark, copyright, blurry, nsfw"
|
58 |
|
59 |
+
# we cannot go beyond 3 seconds on the large A10G
|
60 |
+
video = convert_mp4_to_frames(video_in, duration=3)
|
61 |
video_resized = [Image.fromarray(frame).resize((1024, 576)) for frame in video]
|
62 |
video_frames = pipe_xl(prompt, negative_prompt=negative_prompt, video=video_resized, strength=denoise_strength).frames
|
63 |
video_path = export_to_video(video_frames, output_video_path="xl_result.mp4")
|