JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2026-05-07 15:18:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 9869560fa33cb204bc0b9cea3443a261751901f4 into 7a1af7154511e0ce4e4be8d62faa8c5e5a3532d2

Browse Source
This commit is contained in:
IMaloney 2026-02-18 23:59:19 -05:00 committed by GitHub
commit 24fd32e624
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference/cli_vae_demo.py
View File

@ -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, save it to a file at {encoded_output}/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)