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