"""
Copied from cosmos repository, with necessary modifications.
Original source: https://github.com/nvidia-cosmos/cosmos-predict2.5/
"""