diff options
| author | MalumaDev <piano.lu92@gmail.com> | 2022-10-19 08:22:39 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-19 08:22:39 +0000 |
| commit | 2362d5f00eeecbe6a6f67fe6085da01a3d78e407 (patch) | |
| tree | 763d0a836b63cc0155c4ca810f12db3af57eb448 /modules/hypernetworks | |
| parent | c2765c9bcd264f5a8922348cd03521cb2ff306b3 (diff) | |
| parent | 1b91cbbc1163d3613aa329bed3aecd8e29ca52ca (diff) | |
| download | stable-diffusion-webui-gfx803-2362d5f00eeecbe6a6f67fe6085da01a3d78e407.tar.gz stable-diffusion-webui-gfx803-2362d5f00eeecbe6a6f67fe6085da01a3d78e407.tar.bz2 stable-diffusion-webui-gfx803-2362d5f00eeecbe6a6f67fe6085da01a3d78e407.zip | |
Merge branch 'master' into test_resolve_conflicts
Diffstat (limited to 'modules/hypernetworks')
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index 4905710e..b8695fc1 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -196,7 +196,7 @@ def stack_conds(conds): return torch.stack(conds)
-def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log_directory, steps, 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):
+def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log_directory, training_width, training_height, steps, 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):
assert hypernetwork_name, 'hypernetwork not selected'
path = shared.hypernetworks.get(hypernetwork_name, None)
@@ -225,7 +225,7 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log shared.state.textinfo = f"Preparing dataset from {html.escape(data_root)}..."
with torch.autocast("cuda"):
- ds = modules.textual_inversion.dataset.PersonalizedBase(data_root=data_root, width=512, height=512, repeats=shared.opts.training_image_repeats_per_epoch, placeholder_token=hypernetwork_name, model=shared.sd_model, device=devices.device, template_file=template_file, include_cond=True, batch_size=batch_size)
+ ds = modules.textual_inversion.dataset.PersonalizedBase(data_root=data_root, width=training_width, height=training_height, repeats=shared.opts.training_image_repeats_per_epoch, placeholder_token=hypernetwork_name, model=shared.sd_model, device=devices.device, template_file=template_file, include_cond=True, batch_size=batch_size)
if unload:
shared.sd_model.cond_stage_model.to(devices.cpu)
|