ComfyUI/comfy/ldm/omnigen
..
omnigen2.py