aboutsummaryrefslogtreecommitdiffstats
path: root/modules/shared.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-10-12 17:49:47 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-10-12 17:49:47 +0000
commitc3c8eef9fd5a0c8b26319e32ca4a19b56204e6df (patch)
tree56e3cbdbb3b535b87c173a7a71abfcce21578af9 /modules/shared.py
parentcc5803603b8591075542d99ae8596ab5b130a82f (diff)
downloadstable-diffusion-webui-gfx803-c3c8eef9fd5a0c8b26319e32ca4a19b56204e6df.tar.gz
stable-diffusion-webui-gfx803-c3c8eef9fd5a0c8b26319e32ca4a19b56204e6df.tar.bz2
stable-diffusion-webui-gfx803-c3c8eef9fd5a0c8b26319e32ca4a19b56204e6df.zip
train: change filename processing to be more simple and configurable
train: make it possible to make text files with prompts train: rework scheduler so that there's less repeating code in textual inversion and hypernets train: move epochs setting to options
Diffstat (limited to 'modules/shared.py')
-rw-r--r--modules/shared.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py
index 42e99741..e64e69fc 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -231,6 +231,9 @@ options_templates.update(options_section(('system', "System"), {
options_templates.update(options_section(('training', "Training"), {
"unload_models_when_training": OptionInfo(False, "Unload VAE and CLIP from VRAM when training"),
+ "dataset_filename_word_regex": OptionInfo("", "Filename word regex"),
+ "dataset_filename_join_string": OptionInfo(" ", "Filename join string"),
+ "training_image_repeats_per_epoch": OptionInfo(100, "Number of repeats for a single input image per epoch; used only for displaying epoch number", gr.Number, {"precision": 0}),
}))
options_templates.update(options_section(('sd', "Stable Diffusion"), {