diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-12-25 06:11:42 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-25 06:11:42 +0000 |
| commit | c1512ef9aeb2234ccf7757c678e5c6e2b9d561c4 (patch) | |
| tree | 936630b54b10f7bc5b122b300ce59b6951de3e53 /modules/hypernetworks/hypernetwork.py | |
| parent | 8eef9d8e782aa0655241e43f67059aa7bef3bdca (diff) | |
| parent | 5f1dfbbc959855fd90ba80c0c76301d2063772fa (diff) | |
| download | stable-diffusion-webui-gfx803-c1512ef9aeb2234ccf7757c678e5c6e2b9d561c4.tar.gz stable-diffusion-webui-gfx803-c1512ef9aeb2234ccf7757c678e5c6e2b9d561c4.tar.bz2 stable-diffusion-webui-gfx803-c1512ef9aeb2234ccf7757c678e5c6e2b9d561c4.zip | |
Merge pull request #5999 from vladmandic/trainapi
implement train api
Diffstat (limited to 'modules/hypernetworks/hypernetwork.py')
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index c406ffb3..3182ff03 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -378,6 +378,32 @@ def report_statistics(loss_info:dict): print(e)
+def create_hypernetwork(name, enable_sizes, overwrite_old, layer_structure=None, activation_func=None, weight_init=None, add_layer_norm=False, use_dropout=False):
+ # Remove illegal characters from name.
+ name = "".join( x for x in name if (x.isalnum() or x in "._- "))
+
+ fn = os.path.join(shared.cmd_opts.hypernetwork_dir, f"{name}.pt")
+ if not overwrite_old:
+ assert not os.path.exists(fn), f"file {fn} already exists"
+
+ if type(layer_structure) == str:
+ layer_structure = [float(x.strip()) for x in layer_structure.split(",")]
+
+ hypernet = modules.hypernetworks.hypernetwork.Hypernetwork(
+ name=name,
+ enable_sizes=[int(x) for x in enable_sizes],
+ layer_structure=layer_structure,
+ activation_func=activation_func,
+ weight_init=weight_init,
+ add_layer_norm=add_layer_norm,
+ use_dropout=use_dropout,
+ )
+ hypernet.save(fn)
+
+ shared.reload_hypernetworks()
+
+ return fn
+
def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, gradient_step, data_root, log_directory, training_width, training_height, steps, shuffle_tags, tag_drop_out, latent_sampling_method, create_image_every, save_hypernetwork_every, template_file, preview_from_txt2img, preview_prompt, preview_negative_prompt, preview_steps, preview_sampler_index, preview_cfg_scale, preview_seed, preview_width, preview_height):
# images allows training previews to have infotext. Importing it at the top causes a circular import problem.
|