diff options
| author | Keavon Chambers <keavon@keavon.com> | 2022-11-19 18:34:31 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-19 18:34:31 +0000 |
| commit | 2f90496b19cd9c512633742db97b072a7075f017 (patch) | |
| tree | 60d2dddd69172d9b5cf58c8da2bd64c61132f4fa /modules/hypernetworks/hypernetwork.py | |
| parent | a258fd60dbe2d68325339405a2aa72816d06d2fd (diff) | |
| parent | 47a44c7e421b98ca07e92dbf88769b04c9e28f86 (diff) | |
| download | stable-diffusion-webui-gfx803-2f90496b19cd9c512633742db97b072a7075f017.tar.gz stable-diffusion-webui-gfx803-2f90496b19cd9c512633742db97b072a7075f017.tar.bz2 stable-diffusion-webui-gfx803-2f90496b19cd9c512633742db97b072a7075f017.zip | |
Merge branch 'master' into cors-regex
Diffstat (limited to 'modules/hypernetworks/hypernetwork.py')
| -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 7f182712..fbb87dd1 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -12,7 +12,7 @@ import torch import tqdm
from einops import rearrange, repeat
from ldm.util import default
-from modules import devices, processing, sd_models, shared
+from modules import devices, processing, sd_models, shared, sd_samplers
from modules.textual_inversion import textual_inversion
from modules.textual_inversion.learn_schedule import LearnRateScheduler
from torch import einsum
@@ -535,7 +535,7 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log p.prompt = preview_prompt
p.negative_prompt = preview_negative_prompt
p.steps = preview_steps
- p.sampler_index = preview_sampler_index
+ p.sampler_name = sd_samplers.samplers[preview_sampler_index].name
p.cfg_scale = preview_cfg_scale
p.seed = preview_seed
p.width = preview_width
|