├── .gitattributes ├── .gitignore ├── AITemplate ├── AITemplate.py ├── ComfyCode │ └── model_management.py ├── __init__.py ├── ait │ ├── __init__.py │ ├── ait.py │ ├── compile │ │ ├── clip.py │ │ ├── controlnet.py │ │ ├── release.py │ │ ├── unet.py │ │ ├── util.py │ │ └── vae.py │ ├── inference.py │ ├── load.py │ ├── modeling │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── clip.py │ │ ├── controlnet.py │ │ ├── embeddings.py │ │ ├── resnet.py │ │ ├── unet_2d_condition.py │ │ ├── unet_blocks.py │ │ └── vae.py │ ├── module │ │ ├── __init__.py │ │ ├── dtype.py │ │ ├── misc.py │ │ ├── model.py │ │ └── torch_utils.py │ └── util │ │ ├── __init__.py │ │ ├── ckpt_convert.py │ │ ├── mapping │ │ ├── __init__.py │ │ ├── clip.py │ │ ├── controlnet.py │ │ ├── unet.py │ │ └── vae.py │ │ └── torch_dtype_from_str.py ├── clip.py ├── controlnet.py ├── download_pipeline.py ├── modules │ ├── modules.json │ └── place_modules_here ├── test.py ├── unet.py └── vae.py ├── LICENSE ├── README.md ├── __init__.py ├── docs ├── clip.md ├── compile.md ├── compvis.md ├── controlnet.md ├── unet.md └── vae.md └── workflows ├── aitemplate_controlnet.json ├── aitemplate_img2img_unet_vae.json ├── aitemplate_two_pass.json ├── aitemplate_two_pass_two_model.json ├── aitemplate_unet_only.json └── aitemplate_unet_vae.json /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/.gitignore -------------------------------------------------------------------------------- /AITemplate/AITemplate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/AITemplate.py -------------------------------------------------------------------------------- /AITemplate/ComfyCode/model_management.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ComfyCode/model_management.py -------------------------------------------------------------------------------- /AITemplate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /AITemplate/ait/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/__init__.py -------------------------------------------------------------------------------- /AITemplate/ait/ait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/ait.py -------------------------------------------------------------------------------- /AITemplate/ait/compile/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/compile/clip.py -------------------------------------------------------------------------------- /AITemplate/ait/compile/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/compile/controlnet.py -------------------------------------------------------------------------------- /AITemplate/ait/compile/release.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/compile/release.py -------------------------------------------------------------------------------- /AITemplate/ait/compile/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/compile/unet.py -------------------------------------------------------------------------------- /AITemplate/ait/compile/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/compile/util.py -------------------------------------------------------------------------------- /AITemplate/ait/compile/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/compile/vae.py -------------------------------------------------------------------------------- /AITemplate/ait/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/inference.py -------------------------------------------------------------------------------- /AITemplate/ait/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/load.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/__init__.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/attention.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/clip.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/controlnet.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/embeddings.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/resnet.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/unet_2d_condition.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/unet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/unet_blocks.py -------------------------------------------------------------------------------- /AITemplate/ait/modeling/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/modeling/vae.py -------------------------------------------------------------------------------- /AITemplate/ait/module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/module/__init__.py -------------------------------------------------------------------------------- /AITemplate/ait/module/dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/module/dtype.py -------------------------------------------------------------------------------- /AITemplate/ait/module/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/module/misc.py -------------------------------------------------------------------------------- /AITemplate/ait/module/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/module/model.py -------------------------------------------------------------------------------- /AITemplate/ait/module/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/module/torch_utils.py -------------------------------------------------------------------------------- /AITemplate/ait/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/__init__.py -------------------------------------------------------------------------------- /AITemplate/ait/util/ckpt_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/ckpt_convert.py -------------------------------------------------------------------------------- /AITemplate/ait/util/mapping/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/mapping/__init__.py -------------------------------------------------------------------------------- /AITemplate/ait/util/mapping/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/mapping/clip.py -------------------------------------------------------------------------------- /AITemplate/ait/util/mapping/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/mapping/controlnet.py -------------------------------------------------------------------------------- /AITemplate/ait/util/mapping/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/mapping/unet.py -------------------------------------------------------------------------------- /AITemplate/ait/util/mapping/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/mapping/vae.py -------------------------------------------------------------------------------- /AITemplate/ait/util/torch_dtype_from_str.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/ait/util/torch_dtype_from_str.py -------------------------------------------------------------------------------- /AITemplate/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/clip.py -------------------------------------------------------------------------------- /AITemplate/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/controlnet.py -------------------------------------------------------------------------------- /AITemplate/download_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/download_pipeline.py -------------------------------------------------------------------------------- /AITemplate/modules/modules.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/modules/modules.json -------------------------------------------------------------------------------- /AITemplate/modules/place_modules_here: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /AITemplate/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/test.py -------------------------------------------------------------------------------- /AITemplate/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/unet.py -------------------------------------------------------------------------------- /AITemplate/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/AITemplate/vae.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/__init__.py -------------------------------------------------------------------------------- /docs/clip.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/docs/clip.md -------------------------------------------------------------------------------- /docs/compile.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/docs/compile.md -------------------------------------------------------------------------------- /docs/compvis.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/docs/compvis.md -------------------------------------------------------------------------------- /docs/controlnet.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/docs/controlnet.md -------------------------------------------------------------------------------- /docs/unet.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/docs/unet.md -------------------------------------------------------------------------------- /docs/vae.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/docs/vae.md -------------------------------------------------------------------------------- /workflows/aitemplate_controlnet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/workflows/aitemplate_controlnet.json -------------------------------------------------------------------------------- /workflows/aitemplate_img2img_unet_vae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/workflows/aitemplate_img2img_unet_vae.json -------------------------------------------------------------------------------- /workflows/aitemplate_two_pass.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/workflows/aitemplate_two_pass.json -------------------------------------------------------------------------------- /workflows/aitemplate_two_pass_two_model.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/workflows/aitemplate_two_pass_two_model.json -------------------------------------------------------------------------------- /workflows/aitemplate_unet_only.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/workflows/aitemplate_unet_only.json -------------------------------------------------------------------------------- /workflows/aitemplate_unet_vae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FizzleDorf/AIT/HEAD/workflows/aitemplate_unet_vae.json --------------------------------------------------------------------------------