├── .github └── workflows │ └── publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── assets ├── meme.jpg ├── octupusy.jpg ├── teaser.webp └── woman.jpg ├── data ├── __init__.py ├── configs │ └── example.yaml ├── data_utils.py ├── dataset_base.py ├── dataset_info.py ├── distributed_iterable_dataset.py ├── interleave_datasets │ ├── __init__.py │ ├── edit_dataset.py │ └── interleave_t2i_dataset.py ├── parquet_utils.py ├── t2i_dataset.py ├── transforms.py ├── video_utils.py └── vlm_dataset.py ├── example_workflows ├── bagel_image_editing.json ├── bagel_image_editing.png ├── bagel_image_understanding.json ├── bagel_image_understanding.png ├── bagel_text_to_image.json └── bagel_text_to_image.png ├── inferencer.py ├── modeling ├── __init__.py ├── autoencoder.py ├── bagel │ ├── __init__.py │ ├── bagel.py │ ├── modeling_utils.py │ ├── qwen2_navit.py │ └── siglip_navit.py ├── qwen2 │ ├── __init__.py │ ├── configuration_qwen2.py │ ├── modeling_qwen2.py │ ├── tokenization_qwen2.py │ └── tokenization_qwen2_fast.py └── siglip │ ├── __init__.py │ ├── configuration_siglip.py │ ├── convert_siglip_to_hf.py │ ├── image_processing_siglip.py │ ├── modeling_siglip.py │ ├── processing_siglip.py │ └── tokenization_siglip.py ├── nodes.py ├── pyproject.toml └── requirements.txt /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/meme.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/assets/meme.jpg -------------------------------------------------------------------------------- /assets/octupusy.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/assets/octupusy.jpg -------------------------------------------------------------------------------- /assets/teaser.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/assets/teaser.webp -------------------------------------------------------------------------------- /assets/woman.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/assets/woman.jpg -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/configs/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/configs/example.yaml -------------------------------------------------------------------------------- /data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/data_utils.py -------------------------------------------------------------------------------- /data/dataset_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/dataset_base.py -------------------------------------------------------------------------------- /data/dataset_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/dataset_info.py -------------------------------------------------------------------------------- /data/distributed_iterable_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/distributed_iterable_dataset.py -------------------------------------------------------------------------------- /data/interleave_datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/interleave_datasets/__init__.py -------------------------------------------------------------------------------- /data/interleave_datasets/edit_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/interleave_datasets/edit_dataset.py -------------------------------------------------------------------------------- /data/interleave_datasets/interleave_t2i_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/interleave_datasets/interleave_t2i_dataset.py -------------------------------------------------------------------------------- /data/parquet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/parquet_utils.py -------------------------------------------------------------------------------- /data/t2i_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/t2i_dataset.py -------------------------------------------------------------------------------- /data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/transforms.py -------------------------------------------------------------------------------- /data/video_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/video_utils.py -------------------------------------------------------------------------------- /data/vlm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/data/vlm_dataset.py -------------------------------------------------------------------------------- /example_workflows/bagel_image_editing.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/example_workflows/bagel_image_editing.json -------------------------------------------------------------------------------- /example_workflows/bagel_image_editing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/example_workflows/bagel_image_editing.png -------------------------------------------------------------------------------- /example_workflows/bagel_image_understanding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/example_workflows/bagel_image_understanding.json -------------------------------------------------------------------------------- /example_workflows/bagel_image_understanding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/example_workflows/bagel_image_understanding.png -------------------------------------------------------------------------------- /example_workflows/bagel_text_to_image.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/example_workflows/bagel_text_to_image.json -------------------------------------------------------------------------------- /example_workflows/bagel_text_to_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/example_workflows/bagel_text_to_image.png -------------------------------------------------------------------------------- /inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/inferencer.py -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/__init__.py -------------------------------------------------------------------------------- /modeling/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/autoencoder.py -------------------------------------------------------------------------------- /modeling/bagel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/bagel/__init__.py -------------------------------------------------------------------------------- /modeling/bagel/bagel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/bagel/bagel.py -------------------------------------------------------------------------------- /modeling/bagel/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/bagel/modeling_utils.py -------------------------------------------------------------------------------- /modeling/bagel/qwen2_navit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/bagel/qwen2_navit.py -------------------------------------------------------------------------------- /modeling/bagel/siglip_navit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/bagel/siglip_navit.py -------------------------------------------------------------------------------- /modeling/qwen2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/qwen2/__init__.py -------------------------------------------------------------------------------- /modeling/qwen2/configuration_qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/qwen2/configuration_qwen2.py -------------------------------------------------------------------------------- /modeling/qwen2/modeling_qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/qwen2/modeling_qwen2.py -------------------------------------------------------------------------------- /modeling/qwen2/tokenization_qwen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/qwen2/tokenization_qwen2.py -------------------------------------------------------------------------------- /modeling/qwen2/tokenization_qwen2_fast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/qwen2/tokenization_qwen2_fast.py -------------------------------------------------------------------------------- /modeling/siglip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/__init__.py -------------------------------------------------------------------------------- /modeling/siglip/configuration_siglip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/configuration_siglip.py -------------------------------------------------------------------------------- /modeling/siglip/convert_siglip_to_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/convert_siglip_to_hf.py -------------------------------------------------------------------------------- /modeling/siglip/image_processing_siglip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/image_processing_siglip.py -------------------------------------------------------------------------------- /modeling/siglip/modeling_siglip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/modeling_siglip.py -------------------------------------------------------------------------------- /modeling/siglip/processing_siglip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/processing_siglip.py -------------------------------------------------------------------------------- /modeling/siglip/tokenization_siglip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/modeling/siglip/tokenization_siglip.py -------------------------------------------------------------------------------- /nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/nodes.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neverbiasu/ComfyUI-BAGEL/HEAD/requirements.txt --------------------------------------------------------------------------------