mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-03 07:42:15 -08:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #3975 from aria1th/force-push-patch-13

Browse source 
Save/loading AdamW optimizer (for hypernetworks)
This commit is contained in:
AUTOMATIC1111 2022-11-05 16:15:00 +03:00 • committed by GitHub
commit e96c434495
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 51 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/shared.py
View file

@ -320,6 +320,7 @@ options_templates.update(options_section(('system', "System"), {
options_templates.update(options_section(('training', "Training"), {
"unload_models_when_training": OptionInfo(False, "Move VAE and CLIP to RAM when training if possible. Saves VRAM."),
"save_optimizer_state": OptionInfo(False, "Saves Optimizer state as separate *.optim file. Training can be resumed with HN itself and matching optim file."),
"dataset_filename_word_regex": OptionInfo("", "Filename word regex"),
"dataset_filename_join_string": OptionInfo(" ", "Filename join string"),
"training_image_repeats_per_epoch": OptionInfo(1, "Number of repeats for a single input image per epoch; used only for displaying epoch number", gr.Number, {"precision": 0}),