├── .github └── workflows │ └── publish.yml ├── .gitignore ├── FL_ImageCaptionSaver.py ├── FL_KohyaSSAdvConfig.py ├── FL_KohyaSSDatasetConfig.py ├── FL_KohyaSSInitWorkspace.py ├── FL_KohyaSSTrain.py ├── FL_Kohya_EasyTrain.py ├── FL_LoadImagesFromDirectoryPath.py ├── FL_train_core.py ├── FL_train_utils.py ├── README.md ├── WIP ├── FL_SliderLoraAdvConfig.py.WIP ├── FL_SliderLoraCore.py.WIP ├── FL_SliderLoraDatasetConfig.py.WIP ├── FL_SliderLoraInitWorkspace.py.WIP └── FL_SliderLoraTrain.py.WIP ├── __init__.py ├── configs ├── kohya_ss_lora │ ├── lora_sd1_5.json │ └── lora_sdxl.json └── models_config │ ├── clip-vit-large-patch14 │ └── tokenizer_config.json │ ├── stable-diffusion-v1-5 │ ├── 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 │ ├── 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 ├── example └── Example.json ├── hook_kohya_ss_run.py ├── hook_kohya_ss_utils.py ├── pyproject.toml ├── requirements.txt └── web ├── FL_Trainer.js └── appearance.js /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/.gitignore -------------------------------------------------------------------------------- /FL_ImageCaptionSaver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_ImageCaptionSaver.py -------------------------------------------------------------------------------- /FL_KohyaSSAdvConfig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_KohyaSSAdvConfig.py -------------------------------------------------------------------------------- /FL_KohyaSSDatasetConfig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_KohyaSSDatasetConfig.py -------------------------------------------------------------------------------- /FL_KohyaSSInitWorkspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_KohyaSSInitWorkspace.py -------------------------------------------------------------------------------- /FL_KohyaSSTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_KohyaSSTrain.py -------------------------------------------------------------------------------- /FL_Kohya_EasyTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_Kohya_EasyTrain.py -------------------------------------------------------------------------------- /FL_LoadImagesFromDirectoryPath.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_LoadImagesFromDirectoryPath.py -------------------------------------------------------------------------------- /FL_train_core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_train_core.py -------------------------------------------------------------------------------- /FL_train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/FL_train_utils.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/README.md -------------------------------------------------------------------------------- /WIP/FL_SliderLoraAdvConfig.py.WIP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/WIP/FL_SliderLoraAdvConfig.py.WIP -------------------------------------------------------------------------------- /WIP/FL_SliderLoraCore.py.WIP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/WIP/FL_SliderLoraCore.py.WIP -------------------------------------------------------------------------------- /WIP/FL_SliderLoraDatasetConfig.py.WIP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/WIP/FL_SliderLoraDatasetConfig.py.WIP -------------------------------------------------------------------------------- /WIP/FL_SliderLoraInitWorkspace.py.WIP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/WIP/FL_SliderLoraInitWorkspace.py.WIP -------------------------------------------------------------------------------- /WIP/FL_SliderLoraTrain.py.WIP: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/WIP/FL_SliderLoraTrain.py.WIP -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/__init__.py -------------------------------------------------------------------------------- /configs/kohya_ss_lora/lora_sd1_5.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/configs/kohya_ss_lora/lora_sd1_5.json -------------------------------------------------------------------------------- /configs/kohya_ss_lora/lora_sdxl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/configs/kohya_ss_lora/lora_sdxl.json -------------------------------------------------------------------------------- /configs/models_config/clip-vit-large-patch14/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/configs/models_config/clip-vit-large-patch14/tokenizer_config.json -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-v1-5/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/HEAD/configs/models_config/stable-diffusion-v1.5/v1-inference.yaml -------------------------------------------------------------------------------- /configs/models_config/stable-diffusion-xl-base-1.0/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/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/filliptm/ComfyUI_FL-Trainer/HEAD/configs/models_config/stable-diffusion-xl/sd_xl_base.yaml -------------------------------------------------------------------------------- /example/Example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/example/Example.json -------------------------------------------------------------------------------- /hook_kohya_ss_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/hook_kohya_ss_run.py -------------------------------------------------------------------------------- /hook_kohya_ss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/hook_kohya_ss_utils.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | opencv-python -------------------------------------------------------------------------------- /web/FL_Trainer.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/web/FL_Trainer.js -------------------------------------------------------------------------------- /web/appearance.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/filliptm/ComfyUI_FL-Trainer/HEAD/web/appearance.js --------------------------------------------------------------------------------