├── InstantIR_node.py ├── LICENSE ├── README.md ├── __init__.py ├── assets ├── Konan.png ├── Naruto.png ├── cottage.png ├── dog.png ├── lady.png ├── man.png ├── panda.png ├── sculpture.png └── teaser_figure.png ├── config_files ├── IR_dataset.yaml ├── losses.yaml ├── sd_xl_base.yaml ├── sdxl_repo │ ├── model_index.json │ ├── scheduler │ │ └── scheduler_config.json │ ├── text_encoder │ │ └── config.json │ ├── text_encoder_2 │ │ └── config.json │ ├── tokenizer │ │ ├── merges.txt │ │ ├── special_tokens_map.json │ │ ├── tokenizer_config.json │ │ └── vocab.json │ ├── tokenizer_2 │ │ ├── merges.txt │ │ ├── special_tokens_map.json │ │ ├── tokenizer_config.json │ │ └── vocab.json │ ├── unet │ │ └── config.json │ ├── vae │ │ └── config.json │ ├── vae_1_0 │ │ └── config.json │ ├── vae_decoder │ │ └── config.json │ └── vae_encoder │ │ └── config.json └── val_dataset.yaml ├── data ├── data_config.py └── dataset.py ├── example.png ├── losses ├── loss_config.py └── losses.py ├── module ├── aggregator.py ├── attention.py ├── diffusers_vae │ ├── autoencoder_kl.py │ └── vae.py ├── ip_adapter │ ├── attention_processor.py │ ├── ip_adapter.py │ ├── resampler.py │ └── utils.py ├── min_sdxl.py └── unet │ ├── unet_2d_ZeroSFT.py │ └── unet_2d_ZeroSFT_blocks.py ├── pipelines ├── sdxl_instantir.py └── stage1_sdxl_pipeline.py ├── pyproject.toml ├── requirements.txt ├── schedulers └── lcm_single_step_scheduler.py ├── utils.py ├── utils ├── degradation_pipeline.py ├── matlab_cp2tform.py ├── parser.py ├── text_utils.py ├── train_utils.py ├── utils.py └── vis_utils.py └── workflow.json /InstantIR_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/InstantIR_node.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/Konan.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/Konan.png -------------------------------------------------------------------------------- /assets/Naruto.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/Naruto.png -------------------------------------------------------------------------------- /assets/cottage.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/cottage.png -------------------------------------------------------------------------------- /assets/dog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/dog.png -------------------------------------------------------------------------------- /assets/lady.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/lady.png -------------------------------------------------------------------------------- /assets/man.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/man.png -------------------------------------------------------------------------------- /assets/panda.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/panda.png -------------------------------------------------------------------------------- /assets/sculpture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/sculpture.png -------------------------------------------------------------------------------- /assets/teaser_figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/assets/teaser_figure.png -------------------------------------------------------------------------------- /config_files/IR_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/IR_dataset.yaml -------------------------------------------------------------------------------- /config_files/losses.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/losses.yaml -------------------------------------------------------------------------------- /config_files/sd_xl_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sd_xl_base.yaml -------------------------------------------------------------------------------- /config_files/sdxl_repo/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/model_index.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/scheduler/scheduler_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/scheduler/scheduler_config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/text_encoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/text_encoder/config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/text_encoder_2/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/text_encoder_2/config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer/merges.txt -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer/special_tokens_map.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer/tokenizer_config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer/vocab.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer_2/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer_2/merges.txt -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer_2/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer_2/special_tokens_map.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer_2/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer_2/tokenizer_config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/tokenizer_2/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/tokenizer_2/vocab.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/unet/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/unet/config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/vae/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/vae/config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/vae_1_0/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/vae_1_0/config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/vae_decoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/vae_decoder/config.json -------------------------------------------------------------------------------- /config_files/sdxl_repo/vae_encoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/sdxl_repo/vae_encoder/config.json -------------------------------------------------------------------------------- /config_files/val_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/config_files/val_dataset.yaml -------------------------------------------------------------------------------- /data/data_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/data/data_config.py -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/data/dataset.py -------------------------------------------------------------------------------- /example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/example.png -------------------------------------------------------------------------------- /losses/loss_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/losses/loss_config.py -------------------------------------------------------------------------------- /losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/losses/losses.py -------------------------------------------------------------------------------- /module/aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/aggregator.py -------------------------------------------------------------------------------- /module/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/attention.py -------------------------------------------------------------------------------- /module/diffusers_vae/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/diffusers_vae/autoencoder_kl.py -------------------------------------------------------------------------------- /module/diffusers_vae/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/diffusers_vae/vae.py -------------------------------------------------------------------------------- /module/ip_adapter/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/ip_adapter/attention_processor.py -------------------------------------------------------------------------------- /module/ip_adapter/ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/ip_adapter/ip_adapter.py -------------------------------------------------------------------------------- /module/ip_adapter/resampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/ip_adapter/resampler.py -------------------------------------------------------------------------------- /module/ip_adapter/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/ip_adapter/utils.py -------------------------------------------------------------------------------- /module/min_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/min_sdxl.py -------------------------------------------------------------------------------- /module/unet/unet_2d_ZeroSFT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/unet/unet_2d_ZeroSFT.py -------------------------------------------------------------------------------- /module/unet/unet_2d_ZeroSFT_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/module/unet/unet_2d_ZeroSFT_blocks.py -------------------------------------------------------------------------------- /pipelines/sdxl_instantir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/pipelines/sdxl_instantir.py -------------------------------------------------------------------------------- /pipelines/stage1_sdxl_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/pipelines/stage1_sdxl_pipeline.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/requirements.txt -------------------------------------------------------------------------------- /schedulers/lcm_single_step_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/schedulers/lcm_single_step_scheduler.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils.py -------------------------------------------------------------------------------- /utils/degradation_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/degradation_pipeline.py -------------------------------------------------------------------------------- /utils/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/matlab_cp2tform.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/text_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/text_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/utils/vis_utils.py -------------------------------------------------------------------------------- /workflow.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smthemex/ComfyUI_InstantIR_Wrapper/HEAD/workflow.json --------------------------------------------------------------------------------