├── README.md ├── assets ├── controlnet.png ├── instantid.png ├── lora.png ├── style.png └── teaser.png ├── example ├── Hermione Granger.jpg ├── Johansson.jpg ├── TaylorSwift.png ├── chris-evans.jpg ├── kaifu_resize.png ├── musk_resize.jpeg ├── pose.png ├── pose_rgb.png ├── sam_resize.png ├── schmidhuber_resize.png └── yann-lecun_resize.jpg ├── gradio_demo ├── app.py ├── app_generateOne.py ├── app_instantID.py └── character_template.py ├── inference_instantid.py ├── inference_lora.py ├── requirements.txt └── src ├── efficientvit ├── __init__.py ├── apps │ ├── __init__.py │ ├── data_provider │ │ ├── __init__.py │ │ ├── augment │ │ │ ├── __init__.py │ │ │ ├── bbox.py │ │ │ └── color_aug.py │ │ ├── base.py │ │ └── random_resolution │ │ │ ├── __init__.py │ │ │ ├── _data_loader.py │ │ │ ├── _data_worker.py │ │ │ └── controller.py │ ├── setup.py │ ├── trainer │ │ ├── __init__.py │ │ ├── base.py │ │ └── run_config.py │ └── utils │ │ ├── __init__.py │ │ ├── dist.py │ │ ├── ema.py │ │ ├── export.py │ │ ├── init.py │ │ ├── lr.py │ │ ├── metric.py │ │ ├── misc.py │ │ └── opt.py ├── models │ ├── __init__.py │ ├── efficientvit │ │ ├── __init__.py │ │ ├── backbone.py │ │ ├── cls.py │ │ ├── sam.py │ │ └── seg.py │ ├── nn │ │ ├── __init__.py │ │ ├── act.py │ │ ├── drop.py │ │ ├── norm.py │ │ └── ops.py │ └── utils │ │ ├── __init__.py │ │ ├── list.py │ │ ├── network.py │ │ └── random.py └── sam_model_zoo.py ├── ip_adapter ├── attention_processor.py ├── resampler.py └── utils.py ├── pipelines ├── instantid_pipeline.py ├── instantid_single_pieline.py └── lora_pipeline.py └── prompt_attention ├── p2p_attention.py ├── p2p_utils.py └── seq_aligner.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/README.md -------------------------------------------------------------------------------- /assets/controlnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/assets/controlnet.png -------------------------------------------------------------------------------- /assets/instantid.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/assets/instantid.png -------------------------------------------------------------------------------- /assets/lora.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/assets/lora.png -------------------------------------------------------------------------------- /assets/style.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/assets/style.png -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /example/Hermione Granger.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/Hermione Granger.jpg -------------------------------------------------------------------------------- /example/Johansson.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/Johansson.jpg -------------------------------------------------------------------------------- /example/TaylorSwift.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/TaylorSwift.png -------------------------------------------------------------------------------- /example/chris-evans.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/chris-evans.jpg -------------------------------------------------------------------------------- /example/kaifu_resize.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/kaifu_resize.png -------------------------------------------------------------------------------- /example/musk_resize.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/musk_resize.jpeg -------------------------------------------------------------------------------- /example/pose.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/pose.png -------------------------------------------------------------------------------- /example/pose_rgb.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/pose_rgb.png -------------------------------------------------------------------------------- /example/sam_resize.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/sam_resize.png -------------------------------------------------------------------------------- /example/schmidhuber_resize.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/schmidhuber_resize.png -------------------------------------------------------------------------------- /example/yann-lecun_resize.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/example/yann-lecun_resize.jpg -------------------------------------------------------------------------------- /gradio_demo/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/gradio_demo/app.py -------------------------------------------------------------------------------- /gradio_demo/app_generateOne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/gradio_demo/app_generateOne.py -------------------------------------------------------------------------------- /gradio_demo/app_instantID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/gradio_demo/app_instantID.py -------------------------------------------------------------------------------- /gradio_demo/character_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/gradio_demo/character_template.py -------------------------------------------------------------------------------- /inference_instantid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/inference_instantid.py -------------------------------------------------------------------------------- /inference_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/inference_lora.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/efficientvit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/efficientvit/apps/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/augment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/augment/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/augment/bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/augment/bbox.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/augment/color_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/augment/color_aug.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/base.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/random_resolution/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/random_resolution/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/random_resolution/_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/random_resolution/_data_loader.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/random_resolution/_data_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/random_resolution/_data_worker.py -------------------------------------------------------------------------------- /src/efficientvit/apps/data_provider/random_resolution/controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/data_provider/random_resolution/controller.py -------------------------------------------------------------------------------- /src/efficientvit/apps/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/setup.py -------------------------------------------------------------------------------- /src/efficientvit/apps/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/trainer/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/apps/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/trainer/base.py -------------------------------------------------------------------------------- /src/efficientvit/apps/trainer/run_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/trainer/run_config.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/dist.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/ema.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/export.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/init.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/lr.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/metric.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/misc.py -------------------------------------------------------------------------------- /src/efficientvit/apps/utils/opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/apps/utils/opt.py -------------------------------------------------------------------------------- /src/efficientvit/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/efficientvit/models/efficientvit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/efficientvit/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/models/efficientvit/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/efficientvit/backbone.py -------------------------------------------------------------------------------- /src/efficientvit/models/efficientvit/cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/efficientvit/cls.py -------------------------------------------------------------------------------- /src/efficientvit/models/efficientvit/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/efficientvit/sam.py -------------------------------------------------------------------------------- /src/efficientvit/models/efficientvit/seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/efficientvit/seg.py -------------------------------------------------------------------------------- /src/efficientvit/models/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/nn/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/models/nn/act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/nn/act.py -------------------------------------------------------------------------------- /src/efficientvit/models/nn/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/nn/drop.py -------------------------------------------------------------------------------- /src/efficientvit/models/nn/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/nn/norm.py -------------------------------------------------------------------------------- /src/efficientvit/models/nn/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/nn/ops.py -------------------------------------------------------------------------------- /src/efficientvit/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/utils/__init__.py -------------------------------------------------------------------------------- /src/efficientvit/models/utils/list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/utils/list.py -------------------------------------------------------------------------------- /src/efficientvit/models/utils/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/utils/network.py -------------------------------------------------------------------------------- /src/efficientvit/models/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/models/utils/random.py -------------------------------------------------------------------------------- /src/efficientvit/sam_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/efficientvit/sam_model_zoo.py -------------------------------------------------------------------------------- /src/ip_adapter/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/ip_adapter/attention_processor.py -------------------------------------------------------------------------------- /src/ip_adapter/resampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/ip_adapter/resampler.py -------------------------------------------------------------------------------- /src/ip_adapter/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/ip_adapter/utils.py -------------------------------------------------------------------------------- /src/pipelines/instantid_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/pipelines/instantid_pipeline.py -------------------------------------------------------------------------------- /src/pipelines/instantid_single_pieline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/pipelines/instantid_single_pieline.py -------------------------------------------------------------------------------- /src/pipelines/lora_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/pipelines/lora_pipeline.py -------------------------------------------------------------------------------- /src/prompt_attention/p2p_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/prompt_attention/p2p_attention.py -------------------------------------------------------------------------------- /src/prompt_attention/p2p_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/prompt_attention/p2p_utils.py -------------------------------------------------------------------------------- /src/prompt_attention/seq_aligner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kongzhecn/OMG/HEAD/src/prompt_attention/seq_aligner.py --------------------------------------------------------------------------------