(frames, save_path, fps=7)
| 79 | |
| 80 | |
| 81 | def save_to_mp4(frames, save_path, fps=7): |
| 82 | frames = frames.permute((0, 2, 3, 1)) # (f, c, h, w) to (f, h, w, c) |
| 83 | Path(save_path).parent.mkdir(parents=True, exist_ok=True) |
| 84 | write_video(save_path, frames, fps=fps) |
| 85 | |
| 86 | def faceid_loss_compute(vae, latents, target_images, num_frames, decode_chunk_size, image_processor, device, face_loss_model=None, face_loss_helper=None): |
| 87 |
nothing calls this directly
no outgoing calls
no test coverage detected