├── .github └── workflows │ └── publish.yml ├── .gitignore ├── README.md ├── __init__.py ├── configs ├── kohya_ss_lora │ ├── controlnet_sd1_5.json │ ├── lora_hunyuan1_1.json │ ├── lora_hunyuan1_2.json │ ├── lora_sd1_5.json │ └── lora_sdxl.json ├── models_config │ ├── clip-vit-large-patch14 │ │ └── tokenizer_config.json │ ├── stable-diffusion-v1-5 │ │ ├── .gitattributes │ │ ├── README.md │ │ ├── feature_extractor │ │ │ └── preprocessor_config.json │ │ ├── model_index.json │ │ ├── safety_checker │ │ │ └── config.json │ │ ├── scheduler │ │ │ └── scheduler_config.json │ │ ├── text_encoder │ │ │ └── config.json │ │ ├── tokenizer │ │ │ ├── merges.txt │ │ │ ├── special_tokens_map.json │ │ │ ├── tokenizer_config.json │ │ │ └── vocab.json │ │ ├── unet │ │ │ └── config.json │ │ ├── v1-inference.yaml │ │ └── vae │ │ │ └── config.json │ ├── stable-diffusion-v1.5 │ │ └── v1-inference.yaml │ ├── stable-diffusion-xl-base-1.0 │ │ ├── .gitattributes │ │ ├── LICENSE.md │ │ ├── README.md │ │ ├── model_index.json │ │ ├── scheduler │ │ │ └── scheduler_config.json │ │ ├── text_encoder │ │ │ ├── config.json │ │ │ └── openvino_model.xml │ │ ├── text_encoder_2 │ │ │ ├── config.json │ │ │ └── openvino_model.xml │ │ ├── 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 │ │ │ └── openvino_model.xml │ │ ├── vae │ │ │ └── config.json │ │ ├── vae_1_0 │ │ │ └── config.json │ │ ├── vae_decoder │ │ │ ├── config.json │ │ │ └── openvino_model.xml │ │ └── vae_encoder │ │ │ ├── config.json │ │ │ └── openvino_model.xml │ └── stable-diffusion-xl │ │ └── sd_xl_base.yaml └── sliders │ ├── config-xl.yaml │ └── prompts-xl.yaml ├── examples ├── captioner.png └── workflow.png ├── hook_HYDiT_idk_run.py ├── hook_HYDiT_main_train_deepspeed.py ├── hook_HYDiT_run.py ├── hook_HYDiT_utils.py ├── hook_kohya_ss_hunyuan_pipe.py ├── hook_kohya_ss_run.py ├── hook_kohya_ss_utils.py ├── mz_train_tools_core.py ├── mz_train_tools_core_HYDiT.py ├── mz_train_tools_utils.py └── pyproject.toml /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/__init__.py -------------------------------------------------------------------------------- /configs/kohya_ss_lora/controlnet_sd1_5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/kohya_ss_lora/controlnet_sd1_5.json -------------------------------------------------------------------------------- /configs/kohya_ss_lora/lora_hunyuan1_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/kohya_ss_lora/lora_hunyuan1_1.json -------------------------------------------------------------------------------- /configs/kohya_ss_lora/lora_hunyuan1_2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/kohya_ss_lora/lora_hunyuan1_2.json -------------------------------------------------------------------------------- /configs/kohya_ss_lora/lora_sd1_5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/kohya_ss_lora/lora_sd1_5.json -------------------------------------------------------------------------------- /configs/kohya_ss_lora/lora_sdxl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/kohya_ss_lora/lora_sdxl.json -------------------------------------------------------------------------------- /configs/models_config/clip-vit-large-patch14/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/clip-vit-large-patch14/tokenizer_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/.gitattributes -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/README.md -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/feature_extractor/preprocessor_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/feature_extractor/preprocessor_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/model_index.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/safety_checker/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/safety_checker/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/scheduler/scheduler_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/scheduler/scheduler_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/text_encoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/text_encoder/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/tokenizer/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/tokenizer/merges.txt -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/tokenizer/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/tokenizer/special_tokens_map.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/tokenizer/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/tokenizer/tokenizer_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/tokenizer/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/tokenizer/vocab.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/unet/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/unet/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/v1-inference.yaml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/vae/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1-5/vae/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1.5/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-v1.5/v1-inference.yaml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/.gitattributes -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/LICENSE.md -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/README.md -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/model_index.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/scheduler/scheduler_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/scheduler/scheduler_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/text_encoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/text_encoder/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/text_encoder/openvino_model.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/text_encoder/openvino_model.xml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/text_encoder_2/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/text_encoder_2/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/text_encoder_2/openvino_model.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/text_encoder_2/openvino_model.xml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/merges.txt -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/special_tokens_map.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/tokenizer_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer/vocab.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/merges.txt -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/special_tokens_map.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/tokenizer_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/tokenizer_2/vocab.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/unet/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/unet/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/unet/openvino_model.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/unet/openvino_model.xml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/vae/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/vae/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/vae_1_0/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/vae_1_0/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/vae_decoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/vae_decoder/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/vae_decoder/openvino_model.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/vae_decoder/openvino_model.xml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/vae_encoder/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/vae_encoder/config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/vae_encoder/openvino_model.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl-base-1.0/vae_encoder/openvino_model.xml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl/sd_xl_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/models_config/stable-diffusion-xl/sd_xl_base.yaml -------------------------------------------------------------------------------- /configs/sliders/config-xl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/sliders/config-xl.yaml -------------------------------------------------------------------------------- /configs/sliders/prompts-xl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/configs/sliders/prompts-xl.yaml -------------------------------------------------------------------------------- /examples/captioner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/examples/captioner.png -------------------------------------------------------------------------------- /examples/workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/examples/workflow.png -------------------------------------------------------------------------------- /hook_HYDiT_idk_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_HYDiT_idk_run.py -------------------------------------------------------------------------------- /hook_HYDiT_main_train_deepspeed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_HYDiT_main_train_deepspeed.py -------------------------------------------------------------------------------- /hook_HYDiT_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_HYDiT_run.py -------------------------------------------------------------------------------- /hook_HYDiT_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_HYDiT_utils.py -------------------------------------------------------------------------------- /hook_kohya_ss_hunyuan_pipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_kohya_ss_hunyuan_pipe.py -------------------------------------------------------------------------------- /hook_kohya_ss_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_kohya_ss_run.py -------------------------------------------------------------------------------- /hook_kohya_ss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/hook_kohya_ss_utils.py -------------------------------------------------------------------------------- /mz_train_tools_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/mz_train_tools_core.py -------------------------------------------------------------------------------- /mz_train_tools_core_HYDiT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/mz_train_tools_core_HYDiT.py -------------------------------------------------------------------------------- /mz_train_tools_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/mz_train_tools_utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MinusZoneAI/ComfyUI-TrainTools-MZ/HEAD/pyproject.toml --------------------------------------------------------------------------------