├── Dockerfile ├── README.md ├── __init__.py ├── assets ├── gitconfig-cn └── tensorboard-example.png ├── config ├── autosave │ ├── 20231202-135432.toml │ ├── 20231202-141510.toml │ ├── 20231202-142908.toml │ ├── 20231202-171048.toml │ ├── 20231202-184438.toml │ ├── 20231202-191911.toml │ ├── 20231202-204059.toml │ ├── 20231203-092753.toml │ ├── 20231203-101716.toml │ ├── 20240102-170849.toml │ ├── 20240108-145324.toml │ ├── 20240108-145405.toml │ └── 20240108-151106.toml ├── default.toml ├── lora.toml └── sample_prompts.txt ├── huggingface ├── accelerate │ └── default_config.yaml └── hub │ ├── models--openai--clip-vit-large-patch14 │ ├── .no_exist │ │ └── 32bd64288804d66eefd0ccbe215aa642df71cc41 │ │ │ └── added_tokens.json │ ├── blobs │ │ ├── 4297ea6a8d2bae1fea8f48b45e257814dcb11f69 │ │ ├── 702bb12920b291cade3706cf215c1604d2255d93 │ │ ├── 76e821f1b6f0a9709293c3b6b51ed90980b3166b │ │ └── 9bfb42aa97dcd61e89f279ccaee988bccb4fabae │ ├── refs │ │ └── main │ └── snapshots │ │ └── 32bd64288804d66eefd0ccbe215aa642df71cc41 │ │ ├── merges.txt │ │ ├── special_tokens_map.json │ │ ├── tokenizer_config.json │ │ └── vocab.json │ ├── version.txt │ └── version_diffusers_cache.txt ├── logs ├── .keep ├── 20231202135441 │ └── network_train │ │ └── events.out.tfevents.1701521734.Wendy.12672.0 ├── 20231202141517 │ └── network_train │ │ └── events.out.tfevents.1701522962.Wendy.11456.0 ├── 20231202142915 │ └── network_train │ │ └── events.out.tfevents.1701523799.Wendy.17096.0 ├── 20231202171054 │ └── network_train │ │ └── events.out.tfevents.1701533510.Wendy.3440.0 ├── 20231202184447 │ └── network_train │ │ └── events.out.tfevents.1701539145.Wendy.20612.0 ├── 20231202191918 │ └── network_train │ │ └── events.out.tfevents.1701541203.Wendy.11556.0 ├── 20231202204107 │ └── network_train │ │ └── events.out.tfevents.1701546121.Wendy.21488.0 ├── 20231203092802 │ └── network_train │ │ └── events.out.tfevents.1701592137.Wendy.13404.0 ├── 20231203101724 │ └── network_train │ │ └── events.out.tfevents.1701595098.Wendy.132.0 ├── 20240102170857 │ └── network_train │ │ └── events.out.tfevents.1704211810.Wendy.24100.0 ├── 20240108145412 │ └── network_train │ │ └── events.out.tfevents.1704722107.Wendy.10736.0 ├── Pkmn3GTest20240109113712 │ └── network_train │ │ └── events.out.tfevents.1704796675.Wendy.3216.0 ├── Pkmn3GTest20240109113921 │ └── network_train │ │ └── events.out.tfevents.1704796800.Wendy.7140.0 ├── Pkmn3GTest20240109114358 │ └── network_train │ │ └── events.out.tfevents.1704797075.Wendy.24096.0 └── Pkmn3GTest20240109140433 │ └── network_train │ └── events.out.tfevents.1704805518.Wendy.2964.0 ├── requirements.txt ├── requirements_win.txt ├── sd-scripts ├── .github │ ├── dependabot.yml │ └── workflows │ │ └── typos.yml ├── .gitignore ├── LICENSE.md ├── README-ja.md ├── README.md ├── XTI_hijack.py ├── _typos.toml ├── bitsandbytes_windows │ ├── cextension.py │ ├── libbitsandbytes_cpu.dll │ ├── libbitsandbytes_cuda116.dll │ ├── libbitsandbytes_cuda118.dll │ └── main.py ├── docs │ ├── config_README-ja.md │ ├── fine_tune_README_ja.md │ ├── gen_img_README-ja.md │ ├── train_README-ja.md │ ├── train_README-zh.md │ ├── train_db_README-ja.md │ ├── train_db_README-zh.md │ ├── train_lllite_README-ja.md │ ├── train_lllite_README.md │ ├── train_network_README-ja.md │ ├── train_network_README-zh.md │ └── train_ti_README-ja.md ├── fine_tune.py ├── finetune │ ├── blip │ │ ├── blip.py │ │ ├── med.py │ │ ├── med_config.json │ │ └── vit.py │ ├── clean_captions_and_tags.py │ ├── hypernetwork_nai.py │ ├── make_captions.py │ ├── make_captions_by_git.py │ ├── merge_captions_to_metadata.py │ ├── merge_dd_tags_to_metadata.py │ ├── prepare_buckets_latents.py │ └── tag_images_by_wd14_tagger.py ├── gen_img_diffusers.py ├── library │ ├── __init__.py │ ├── attention_processors.py │ ├── config_util.py │ ├── custom_train_functions.py │ ├── huggingface_util.py │ ├── hypernetwork.py │ ├── ipex │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── diffusers.py │ │ ├── gradscaler.py │ │ └── hijacks.py │ ├── lpw_stable_diffusion.py │ ├── model_util.py │ ├── original_unet.py │ ├── sai_model_spec.py │ ├── sdxl_lpw_stable_diffusion.py │ ├── sdxl_model_util.py │ ├── sdxl_original_unet.py │ ├── sdxl_train_util.py │ ├── slicing_vae.py │ ├── train_util.py │ └── utils.py ├── networks │ ├── check_lora_weights.py │ ├── control_net_lllite.py │ ├── control_net_lllite_for_train.py │ ├── dylora.py │ ├── extract_lora_from_dylora.py │ ├── extract_lora_from_models.py │ ├── lora.py │ ├── lora_diffusers.py │ ├── lora_fa.py │ ├── lora_interrogator.py │ ├── merge_lora.py │ ├── merge_lora_old.py │ ├── oft.py │ ├── resize_lora.py │ ├── sdxl_merge_lora.py │ └── svd_merge_lora.py ├── requirements.txt ├── sdxl_gen_img.py ├── sdxl_minimal_inference.py ├── sdxl_train.py ├── sdxl_train_control_net_lllite.py ├── sdxl_train_control_net_lllite_old.py ├── sdxl_train_network.py ├── sdxl_train_textual_inversion.py ├── setup.py ├── tools │ ├── cache_latents.py │ ├── cache_text_encoder_outputs.py │ ├── canny.py │ ├── convert_diffusers20_original_sd.py │ ├── detect_face_rotate.py │ ├── latent_upscaler.py │ ├── merge_models.py │ ├── original_control_net.py │ ├── resize_images_to_resolution.py │ └── show_metadata.py ├── train_controlnet.py ├── train_db.py ├── train_network.py ├── train_textual_inversion.py └── train_textual_inversion_XTI.py ├── tensorboard.ps1 └── train.py /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/Dockerfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/gitconfig-cn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/assets/gitconfig-cn -------------------------------------------------------------------------------- /assets/tensorboard-example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/assets/tensorboard-example.png -------------------------------------------------------------------------------- /config/autosave/20231202-135432.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-135432.toml -------------------------------------------------------------------------------- /config/autosave/20231202-141510.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-141510.toml -------------------------------------------------------------------------------- /config/autosave/20231202-142908.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-142908.toml -------------------------------------------------------------------------------- /config/autosave/20231202-171048.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-171048.toml -------------------------------------------------------------------------------- /config/autosave/20231202-184438.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-184438.toml -------------------------------------------------------------------------------- /config/autosave/20231202-191911.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-191911.toml -------------------------------------------------------------------------------- /config/autosave/20231202-204059.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231202-204059.toml -------------------------------------------------------------------------------- /config/autosave/20231203-092753.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231203-092753.toml -------------------------------------------------------------------------------- /config/autosave/20231203-101716.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20231203-101716.toml -------------------------------------------------------------------------------- /config/autosave/20240102-170849.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20240102-170849.toml -------------------------------------------------------------------------------- /config/autosave/20240108-145324.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20240108-145324.toml -------------------------------------------------------------------------------- /config/autosave/20240108-145405.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20240108-145405.toml -------------------------------------------------------------------------------- /config/autosave/20240108-151106.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/autosave/20240108-151106.toml -------------------------------------------------------------------------------- /config/default.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/default.toml -------------------------------------------------------------------------------- /config/lora.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/lora.toml -------------------------------------------------------------------------------- /config/sample_prompts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/config/sample_prompts.txt -------------------------------------------------------------------------------- /huggingface/accelerate/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/accelerate/default_config.yaml -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/.no_exist/32bd64288804d66eefd0ccbe215aa642df71cc41/added_tokens.json: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/blobs/4297ea6a8d2bae1fea8f48b45e257814dcb11f69: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/blobs/4297ea6a8d2bae1fea8f48b45e257814dcb11f69 -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/blobs/702bb12920b291cade3706cf215c1604d2255d93: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/blobs/702bb12920b291cade3706cf215c1604d2255d93 -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/blobs/76e821f1b6f0a9709293c3b6b51ed90980b3166b: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/blobs/76e821f1b6f0a9709293c3b6b51ed90980b3166b -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/blobs/9bfb42aa97dcd61e89f279ccaee988bccb4fabae: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/blobs/9bfb42aa97dcd61e89f279ccaee988bccb4fabae -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/refs/main: -------------------------------------------------------------------------------- 1 | 32bd64288804d66eefd0ccbe215aa642df71cc41 -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/merges.txt -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/special_tokens_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/special_tokens_map.json -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/tokenizer_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/tokenizer_config.json -------------------------------------------------------------------------------- /huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/huggingface/hub/models--openai--clip-vit-large-patch14/snapshots/32bd64288804d66eefd0ccbe215aa642df71cc41/vocab.json -------------------------------------------------------------------------------- /huggingface/hub/version.txt: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /huggingface/hub/version_diffusers_cache.txt: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /logs/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /logs/20231202135441/network_train/events.out.tfevents.1701521734.Wendy.12672.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202135441/network_train/events.out.tfevents.1701521734.Wendy.12672.0 -------------------------------------------------------------------------------- /logs/20231202141517/network_train/events.out.tfevents.1701522962.Wendy.11456.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202141517/network_train/events.out.tfevents.1701522962.Wendy.11456.0 -------------------------------------------------------------------------------- /logs/20231202142915/network_train/events.out.tfevents.1701523799.Wendy.17096.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202142915/network_train/events.out.tfevents.1701523799.Wendy.17096.0 -------------------------------------------------------------------------------- /logs/20231202171054/network_train/events.out.tfevents.1701533510.Wendy.3440.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202171054/network_train/events.out.tfevents.1701533510.Wendy.3440.0 -------------------------------------------------------------------------------- /logs/20231202184447/network_train/events.out.tfevents.1701539145.Wendy.20612.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202184447/network_train/events.out.tfevents.1701539145.Wendy.20612.0 -------------------------------------------------------------------------------- /logs/20231202191918/network_train/events.out.tfevents.1701541203.Wendy.11556.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202191918/network_train/events.out.tfevents.1701541203.Wendy.11556.0 -------------------------------------------------------------------------------- /logs/20231202204107/network_train/events.out.tfevents.1701546121.Wendy.21488.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231202204107/network_train/events.out.tfevents.1701546121.Wendy.21488.0 -------------------------------------------------------------------------------- /logs/20231203092802/network_train/events.out.tfevents.1701592137.Wendy.13404.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231203092802/network_train/events.out.tfevents.1701592137.Wendy.13404.0 -------------------------------------------------------------------------------- /logs/20231203101724/network_train/events.out.tfevents.1701595098.Wendy.132.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20231203101724/network_train/events.out.tfevents.1701595098.Wendy.132.0 -------------------------------------------------------------------------------- /logs/20240102170857/network_train/events.out.tfevents.1704211810.Wendy.24100.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20240102170857/network_train/events.out.tfevents.1704211810.Wendy.24100.0 -------------------------------------------------------------------------------- /logs/20240108145412/network_train/events.out.tfevents.1704722107.Wendy.10736.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/20240108145412/network_train/events.out.tfevents.1704722107.Wendy.10736.0 -------------------------------------------------------------------------------- /logs/Pkmn3GTest20240109113712/network_train/events.out.tfevents.1704796675.Wendy.3216.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/Pkmn3GTest20240109113712/network_train/events.out.tfevents.1704796675.Wendy.3216.0 -------------------------------------------------------------------------------- /logs/Pkmn3GTest20240109113921/network_train/events.out.tfevents.1704796800.Wendy.7140.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/Pkmn3GTest20240109113921/network_train/events.out.tfevents.1704796800.Wendy.7140.0 -------------------------------------------------------------------------------- /logs/Pkmn3GTest20240109114358/network_train/events.out.tfevents.1704797075.Wendy.24096.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/Pkmn3GTest20240109114358/network_train/events.out.tfevents.1704797075.Wendy.24096.0 -------------------------------------------------------------------------------- /logs/Pkmn3GTest20240109140433/network_train/events.out.tfevents.1704805518.Wendy.2964.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/logs/Pkmn3GTest20240109140433/network_train/events.out.tfevents.1704805518.Wendy.2964.0 -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_win.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/requirements_win.txt -------------------------------------------------------------------------------- /sd-scripts/.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/.github/dependabot.yml -------------------------------------------------------------------------------- /sd-scripts/.github/workflows/typos.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/.github/workflows/typos.yml -------------------------------------------------------------------------------- /sd-scripts/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/.gitignore -------------------------------------------------------------------------------- /sd-scripts/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/LICENSE.md -------------------------------------------------------------------------------- /sd-scripts/README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/README-ja.md -------------------------------------------------------------------------------- /sd-scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/README.md -------------------------------------------------------------------------------- /sd-scripts/XTI_hijack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/XTI_hijack.py -------------------------------------------------------------------------------- /sd-scripts/_typos.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/_typos.toml -------------------------------------------------------------------------------- /sd-scripts/bitsandbytes_windows/cextension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/bitsandbytes_windows/cextension.py -------------------------------------------------------------------------------- /sd-scripts/bitsandbytes_windows/libbitsandbytes_cpu.dll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/bitsandbytes_windows/libbitsandbytes_cpu.dll -------------------------------------------------------------------------------- /sd-scripts/bitsandbytes_windows/libbitsandbytes_cuda116.dll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/bitsandbytes_windows/libbitsandbytes_cuda116.dll -------------------------------------------------------------------------------- /sd-scripts/bitsandbytes_windows/libbitsandbytes_cuda118.dll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/bitsandbytes_windows/libbitsandbytes_cuda118.dll -------------------------------------------------------------------------------- /sd-scripts/bitsandbytes_windows/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/bitsandbytes_windows/main.py -------------------------------------------------------------------------------- /sd-scripts/docs/config_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/config_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/fine_tune_README_ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/fine_tune_README_ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/gen_img_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/gen_img_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_README-zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_README-zh.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_db_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_db_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_db_README-zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_db_README-zh.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_lllite_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_lllite_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_lllite_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_lllite_README.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_network_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_network_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_network_README-zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_network_README-zh.md -------------------------------------------------------------------------------- /sd-scripts/docs/train_ti_README-ja.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/docs/train_ti_README-ja.md -------------------------------------------------------------------------------- /sd-scripts/fine_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/fine_tune.py -------------------------------------------------------------------------------- /sd-scripts/finetune/blip/blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/blip/blip.py -------------------------------------------------------------------------------- /sd-scripts/finetune/blip/med.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/blip/med.py -------------------------------------------------------------------------------- /sd-scripts/finetune/blip/med_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/blip/med_config.json -------------------------------------------------------------------------------- /sd-scripts/finetune/blip/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/blip/vit.py -------------------------------------------------------------------------------- /sd-scripts/finetune/clean_captions_and_tags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/clean_captions_and_tags.py -------------------------------------------------------------------------------- /sd-scripts/finetune/hypernetwork_nai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/hypernetwork_nai.py -------------------------------------------------------------------------------- /sd-scripts/finetune/make_captions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/make_captions.py -------------------------------------------------------------------------------- /sd-scripts/finetune/make_captions_by_git.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/make_captions_by_git.py -------------------------------------------------------------------------------- /sd-scripts/finetune/merge_captions_to_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/merge_captions_to_metadata.py -------------------------------------------------------------------------------- /sd-scripts/finetune/merge_dd_tags_to_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/merge_dd_tags_to_metadata.py -------------------------------------------------------------------------------- /sd-scripts/finetune/prepare_buckets_latents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/prepare_buckets_latents.py -------------------------------------------------------------------------------- /sd-scripts/finetune/tag_images_by_wd14_tagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/finetune/tag_images_by_wd14_tagger.py -------------------------------------------------------------------------------- /sd-scripts/gen_img_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/gen_img_diffusers.py -------------------------------------------------------------------------------- /sd-scripts/library/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sd-scripts/library/attention_processors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/attention_processors.py -------------------------------------------------------------------------------- /sd-scripts/library/config_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/config_util.py -------------------------------------------------------------------------------- /sd-scripts/library/custom_train_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/custom_train_functions.py -------------------------------------------------------------------------------- /sd-scripts/library/huggingface_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/huggingface_util.py -------------------------------------------------------------------------------- /sd-scripts/library/hypernetwork.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/hypernetwork.py -------------------------------------------------------------------------------- /sd-scripts/library/ipex/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/ipex/__init__.py -------------------------------------------------------------------------------- /sd-scripts/library/ipex/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/ipex/attention.py -------------------------------------------------------------------------------- /sd-scripts/library/ipex/diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/ipex/diffusers.py -------------------------------------------------------------------------------- /sd-scripts/library/ipex/gradscaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/ipex/gradscaler.py -------------------------------------------------------------------------------- /sd-scripts/library/ipex/hijacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/ipex/hijacks.py -------------------------------------------------------------------------------- /sd-scripts/library/lpw_stable_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/lpw_stable_diffusion.py -------------------------------------------------------------------------------- /sd-scripts/library/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/model_util.py -------------------------------------------------------------------------------- /sd-scripts/library/original_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/original_unet.py -------------------------------------------------------------------------------- /sd-scripts/library/sai_model_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/sai_model_spec.py -------------------------------------------------------------------------------- /sd-scripts/library/sdxl_lpw_stable_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/sdxl_lpw_stable_diffusion.py -------------------------------------------------------------------------------- /sd-scripts/library/sdxl_model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/sdxl_model_util.py -------------------------------------------------------------------------------- /sd-scripts/library/sdxl_original_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/sdxl_original_unet.py -------------------------------------------------------------------------------- /sd-scripts/library/sdxl_train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/sdxl_train_util.py -------------------------------------------------------------------------------- /sd-scripts/library/slicing_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/slicing_vae.py -------------------------------------------------------------------------------- /sd-scripts/library/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/train_util.py -------------------------------------------------------------------------------- /sd-scripts/library/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/library/utils.py -------------------------------------------------------------------------------- /sd-scripts/networks/check_lora_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/check_lora_weights.py -------------------------------------------------------------------------------- /sd-scripts/networks/control_net_lllite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/control_net_lllite.py -------------------------------------------------------------------------------- /sd-scripts/networks/control_net_lllite_for_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/control_net_lllite_for_train.py -------------------------------------------------------------------------------- /sd-scripts/networks/dylora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/dylora.py -------------------------------------------------------------------------------- /sd-scripts/networks/extract_lora_from_dylora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/extract_lora_from_dylora.py -------------------------------------------------------------------------------- /sd-scripts/networks/extract_lora_from_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/extract_lora_from_models.py -------------------------------------------------------------------------------- /sd-scripts/networks/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/lora.py -------------------------------------------------------------------------------- /sd-scripts/networks/lora_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/lora_diffusers.py -------------------------------------------------------------------------------- /sd-scripts/networks/lora_fa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/lora_fa.py -------------------------------------------------------------------------------- /sd-scripts/networks/lora_interrogator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/lora_interrogator.py -------------------------------------------------------------------------------- /sd-scripts/networks/merge_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/merge_lora.py -------------------------------------------------------------------------------- /sd-scripts/networks/merge_lora_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/merge_lora_old.py -------------------------------------------------------------------------------- /sd-scripts/networks/oft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/oft.py -------------------------------------------------------------------------------- /sd-scripts/networks/resize_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/resize_lora.py -------------------------------------------------------------------------------- /sd-scripts/networks/sdxl_merge_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/sdxl_merge_lora.py -------------------------------------------------------------------------------- /sd-scripts/networks/svd_merge_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/networks/svd_merge_lora.py -------------------------------------------------------------------------------- /sd-scripts/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/requirements.txt -------------------------------------------------------------------------------- /sd-scripts/sdxl_gen_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_gen_img.py -------------------------------------------------------------------------------- /sd-scripts/sdxl_minimal_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_minimal_inference.py -------------------------------------------------------------------------------- /sd-scripts/sdxl_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_train.py -------------------------------------------------------------------------------- /sd-scripts/sdxl_train_control_net_lllite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_train_control_net_lllite.py -------------------------------------------------------------------------------- /sd-scripts/sdxl_train_control_net_lllite_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_train_control_net_lllite_old.py -------------------------------------------------------------------------------- /sd-scripts/sdxl_train_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_train_network.py -------------------------------------------------------------------------------- /sd-scripts/sdxl_train_textual_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/sdxl_train_textual_inversion.py -------------------------------------------------------------------------------- /sd-scripts/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/setup.py -------------------------------------------------------------------------------- /sd-scripts/tools/cache_latents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/cache_latents.py -------------------------------------------------------------------------------- /sd-scripts/tools/cache_text_encoder_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/cache_text_encoder_outputs.py -------------------------------------------------------------------------------- /sd-scripts/tools/canny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/canny.py -------------------------------------------------------------------------------- /sd-scripts/tools/convert_diffusers20_original_sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/convert_diffusers20_original_sd.py -------------------------------------------------------------------------------- /sd-scripts/tools/detect_face_rotate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/detect_face_rotate.py -------------------------------------------------------------------------------- /sd-scripts/tools/latent_upscaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/latent_upscaler.py -------------------------------------------------------------------------------- /sd-scripts/tools/merge_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/merge_models.py -------------------------------------------------------------------------------- /sd-scripts/tools/original_control_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/original_control_net.py -------------------------------------------------------------------------------- /sd-scripts/tools/resize_images_to_resolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/resize_images_to_resolution.py -------------------------------------------------------------------------------- /sd-scripts/tools/show_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/tools/show_metadata.py -------------------------------------------------------------------------------- /sd-scripts/train_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/train_controlnet.py -------------------------------------------------------------------------------- /sd-scripts/train_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/train_db.py -------------------------------------------------------------------------------- /sd-scripts/train_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/train_network.py -------------------------------------------------------------------------------- /sd-scripts/train_textual_inversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/train_textual_inversion.py -------------------------------------------------------------------------------- /sd-scripts/train_textual_inversion_XTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/sd-scripts/train_textual_inversion_XTI.py -------------------------------------------------------------------------------- /tensorboard.ps1: -------------------------------------------------------------------------------- 1 | $Env:TF_CPP_MIN_LOG_LEVEL = "3" 2 | 3 | .\venv\Scripts\activate 4 | tensorboard --logdir=logs -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LarryJane491/Lora-Training-in-Comfy/HEAD/train.py --------------------------------------------------------------------------------