diff options
| author | xucj98 <975114697@qq.com> | 2022-11-25 09:07:00 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-25 09:07:00 +0000 |
| commit | 263b323de12eb2444b0818105575a2bc69ab0344 (patch) | |
| tree | 7bc4a7802bdaba21550ad281cc08c3fc1037b074 /modules/textual_inversion/textual_inversion.py | |
| parent | d20dbe47e06de7f6c0e65242a04c9bb1410ef7cb (diff) | |
| parent | 828438b4a190759807f9054932cae3a8b880ddf1 (diff) | |
| download | stable-diffusion-webui-gfx803-263b323de12eb2444b0818105575a2bc69ab0344.tar.gz stable-diffusion-webui-gfx803-263b323de12eb2444b0818105575a2bc69ab0344.tar.bz2 stable-diffusion-webui-gfx803-263b323de12eb2444b0818105575a2bc69ab0344.zip | |
Merge branch 'AUTOMATIC1111:master' into draft
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 0aeb0459..5e4d8688 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -10,7 +10,7 @@ import csv from PIL import Image, PngImagePlugin
-from modules import shared, devices, sd_hijack, processing, sd_models, images
+from modules import shared, devices, sd_hijack, processing, sd_models, images, sd_samplers
import modules.textual_inversion.dataset
from modules.textual_inversion.learn_schedule import LearnRateScheduler
@@ -345,7 +345,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, data_root, log_direc 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
|