Files
ComfyUI/comfy/ldm/cosmos/predict2.py