diff --git a/inference/cli_vae_demo.py b/inference/cli_vae_demo.py index cb95d82..31c0696 100644 --- a/inference/cli_vae_demo.py +++ b/inference/cli_vae_demo.py @@ -142,9 +142,7 @@ if __name__ == "__main__": 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") - print( - f"Finished encoding the video to a tensor, saved to {args.output_path}/encoded.pt" - ) + print(f"Finished encoding the video to a tensor, saved to {args.output_path}/encoded.pt") elif args.mode == "decode": assert args.encoded_path, "Encoded tensor path must be provided for decoding." decoded_output = decode_video(args.model_path, args.encoded_path, dtype, device)