From d0277060520158ef27bf181c02b8d8d20d01e646 Mon Sep 17 00:00:00 2001 From: Haiyi Date: Tue, 13 Aug 2024 15:38:39 +0800 Subject: [PATCH] Update cli_vae_demo.py --- inference/cli_vae_demo.py | 1 + 1 file changed, 1 insertion(+) diff --git a/inference/cli_vae_demo.py b/inference/cli_vae_demo.py index 18b9a95..2216053 100644 --- a/inference/cli_vae_demo.py +++ b/inference/cli_vae_demo.py @@ -133,5 +133,6 @@ if __name__ == "__main__": elif args.mode == "both": assert args.video_path, "Video path must be provided for encoding." encoded_output = encode_video(args.model_path, args.video_path, dtype, device) + torch.save(encoded_output, args.output_path + "/encoded.pt") decoded_output = decode_video(args.model_path, args.output_path + "/encoded.pt", dtype, device) save_video(decoded_output, args.output_path)