diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-15 10:10:16 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-15 10:10:16 +0000 |
| commit | f2693bec08d2c2e513cb35fa24402396505a01a9 (patch) | |
| tree | e17b80c308467f9c1433bfc632ec7ecd42dae372 /modules/processing.py | |
| parent | b28cf84c3632df4a6d4c110f7c25d68445b64427 (diff) | |
| download | stable-diffusion-webui-gfx803-f2693bec08d2c2e513cb35fa24402396505a01a9.tar.gz stable-diffusion-webui-gfx803-f2693bec08d2c2e513cb35fa24402396505a01a9.tar.bz2 stable-diffusion-webui-gfx803-f2693bec08d2c2e513cb35fa24402396505a01a9.zip | |
prompt editing
Diffstat (limited to 'modules/processing.py')
| -rw-r--r-- | modules/processing.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/modules/processing.py b/modules/processing.py index 93138e7c..9b53d210 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -12,7 +12,7 @@ import cv2 from skimage import exposure
import modules.sd_hijack
-from modules import devices
+from modules import devices, prompt_parser
from modules.sd_hijack import model_hijack
from modules.sd_samplers import samplers, samplers_for_img2img
from modules.shared import opts, cmd_opts, state
@@ -247,8 +247,10 @@ def process_images(p: StableDiffusionProcessing) -> Processed: seeds = all_seeds[n * p.batch_size:(n + 1) * p.batch_size]
subseeds = all_subseeds[n * p.batch_size:(n + 1) * p.batch_size]
- uc = p.sd_model.get_learned_conditioning(len(prompts) * [p.negative_prompt])
- c = p.sd_model.get_learned_conditioning(prompts)
+ #uc = p.sd_model.get_learned_conditioning(len(prompts) * [p.negative_prompt])
+ #c = p.sd_model.get_learned_conditioning(prompts)
+ uc = prompt_parser.get_learned_conditioning(len(prompts) * [p.negative_prompt], p.steps)
+ c = prompt_parser.get_learned_conditioning(prompts, p.steps)
if len(model_hijack.comments) > 0:
for comment in model_hijack.comments:
|