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