1. git clone https://github.com/nvidia-cosmos/cosmos-predict1.git
2. replace cosmos_predict1 folder with cosmos1 folder
3. setup environment following cosmos repo
4. bash run.sh


# Main modifications: cosmos1/models/autoregressive/utils/sampling.py, cosmos1/models/autoregressive/model.py, cosmos1/models/autoregressive/networks/transformer.py