jbilcke-hf HF staff commited on
Commit
327e1e8
1 Parent(s): 0fa742d

out of memory! let's scale down to 3 seconds

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- video = convert_mp4_to_frames(video_in, duration=20)
 
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")