Closed Sidd065 closed 1 month ago
It is possible to do that. I can give you some suggestions. (1) Split the 2000 frames to small subsets, for instance 250 frames per subset. (2) initialize variables subset by subset, such as landmarks, latents and audios. (3) call pipe on each subset. (4) concat the results as the final result.
I am using a driver video with ~2000 frames. Currently in
infer_audio2vid_pose_acc.py
all 2000 .pkl files in pose_dir are loaded at once and whenface_locator
is run I get atorch.cuda.OutOfMemoryError: CUDA out of memory
. Is it possible to batch process the .pkl files and concatenate the resulting videos to reduce VRAM requirement for large videos?