aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin/Lora/ui_extra_networks_lora.py
diff options
context:
space:
mode:
authorLearwin <6223515+Learwin@users.noreply.github.com>2023-12-30 21:52:27 +0100
committerLearwin <6223515+Learwin@users.noreply.github.com>2023-12-30 21:52:27 +0100
commitbc5ae74c7d8949bab37e260b16e76889b9968099 (patch)
tree8bd498b7cceb850cd5a9ba461bf4cd9eba43696b /extensions-builtin/Lora/ui_extra_networks_lora.py
parent8100e901ab0c5b04d289eebb722c8a653b8beef1 (diff)
Added negative prompts to extra networks lora
Diffstat (limited to 'extensions-builtin/Lora/ui_extra_networks_lora.py')
-rw-r--r--extensions-builtin/Lora/ui_extra_networks_lora.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/ui_extra_networks_lora.py b/extensions-builtin/Lora/ui_extra_networks_lora.py
index df02c663..09ce2a05 100644
--- a/extensions-builtin/Lora/ui_extra_networks_lora.py
+++ b/extensions-builtin/Lora/ui_extra_networks_lora.py
@@ -45,6 +45,15 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
if activation_text:
item["prompt"] += " + " + quote_js(" " + activation_text)
+ negative_prompt = item["user_metadata"].get("negative text")
+ preferred_negative_weight = item["user_metadata"].get("negative weight")
+ item["negative_prompt"] = quote_js("")
+ if negative_prompt:
+ neg_prompt = negative_prompt
+ if (preferred_negative_weight > 0):
+ neg_prompt = '(' + negative_prompt + ':' + str(preferred_negative_weight) + ')'
+ item["negative_prompt"] = quote_js(neg_prompt)
+
sd_version = item["user_metadata"].get("sd version")
if sd_version in network.SdVersion.__members__:
item["sd_version"] = sd_version