Merge pull request #2 from InvincibleDude/extra-networks-test

Extra networks test
This commit is contained in:
InvincibleDude 2023-01-29 20:40:06 +03:00 committed by GitHub
commit 0d834b9394
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -555,6 +555,12 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
model_hijack.embedding_db.load_textual_inversion_embeddings()
_, extra_network_data = extra_networks.parse_prompts(p.all_prompts[0:1])
if type(p) == StableDiffusionProcessingTxt2Img:
if p.enable_hr and p.hr_prompt != '':
_, hr_extra_network_data = extra_networks.parse_prompts(p.all_hr_prompts[0:1])
if p.all_hr_prompts != p.all_prompts:
extra_network_data.update(hr_extra_network_data)
if p.scripts is not None:
p.scripts.process(p)