├── LICENSE ├── README.md ├── config ├── README.md └── example.yaml ├── main.py ├── modules ├── config.py ├── controlnet │ └── canny_dataset.py ├── dataset.py ├── diffusion_model.py ├── lcm │ └── lcm_trainer.py ├── scheduler.py ├── text_model.py ├── trainer.py └── utils.py ├── networks ├── lora.py ├── lortnoc │ └── manager.py └── manager.py ├── preprocess ├── bucketing.py ├── caption_preprocessor.py ├── clip_vision.py ├── create_mask.py ├── create_pfg_feature.py ├── dbimutils.py ├── latent.py ├── make_metadata.py ├── make_original_size_data.py ├── model.py ├── tagger.py ├── text_embedding.py └── utils.py ├── requirements.txt └── tools ├── convert_controlnet.py ├── convert_lora_sdxl.py └── create_control_lora.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/README.md -------------------------------------------------------------------------------- /config/README.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /config/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/config/example.yaml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/main.py -------------------------------------------------------------------------------- /modules/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/config.py -------------------------------------------------------------------------------- /modules/controlnet/canny_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/controlnet/canny_dataset.py -------------------------------------------------------------------------------- /modules/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/dataset.py -------------------------------------------------------------------------------- /modules/diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/diffusion_model.py -------------------------------------------------------------------------------- /modules/lcm/lcm_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/lcm/lcm_trainer.py -------------------------------------------------------------------------------- /modules/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/scheduler.py -------------------------------------------------------------------------------- /modules/text_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/text_model.py -------------------------------------------------------------------------------- /modules/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/trainer.py -------------------------------------------------------------------------------- /modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/modules/utils.py -------------------------------------------------------------------------------- /networks/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/networks/lora.py -------------------------------------------------------------------------------- /networks/lortnoc/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/networks/lortnoc/manager.py -------------------------------------------------------------------------------- /networks/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/networks/manager.py -------------------------------------------------------------------------------- /preprocess/bucketing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/bucketing.py -------------------------------------------------------------------------------- /preprocess/caption_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/caption_preprocessor.py -------------------------------------------------------------------------------- /preprocess/clip_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/clip_vision.py -------------------------------------------------------------------------------- /preprocess/create_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/create_mask.py -------------------------------------------------------------------------------- /preprocess/create_pfg_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/create_pfg_feature.py -------------------------------------------------------------------------------- /preprocess/dbimutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/dbimutils.py -------------------------------------------------------------------------------- /preprocess/latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/latent.py -------------------------------------------------------------------------------- /preprocess/make_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/make_metadata.py -------------------------------------------------------------------------------- /preprocess/make_original_size_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/make_original_size_data.py -------------------------------------------------------------------------------- /preprocess/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/model.py -------------------------------------------------------------------------------- /preprocess/tagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/tagger.py -------------------------------------------------------------------------------- /preprocess/text_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/text_embedding.py -------------------------------------------------------------------------------- /preprocess/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/preprocess/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/convert_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/tools/convert_controlnet.py -------------------------------------------------------------------------------- /tools/convert_lora_sdxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/tools/convert_lora_sdxl.py -------------------------------------------------------------------------------- /tools/create_control_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/laksjdjf/sd-trainer/HEAD/tools/create_control_lora.py --------------------------------------------------------------------------------