diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-25 11:29:46 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-25 11:29:46 +0300 |
| commit | 1bfec873fa13d803f3d4ac2a12bf6983838233fe (patch) | |
| tree | a51fc3389fd2165397ef408d1402322e57331a65 /extensions-builtin/Lora/scripts | |
| parent | 48a15821de768fea76e66f26df83df3fddf18f4b (diff) | |
add an experimental option to apply loras to outputs rather than inputs
Diffstat (limited to 'extensions-builtin/Lora/scripts')
| -rw-r--r-- | extensions-builtin/Lora/scripts/lora_script.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/scripts/lora_script.py b/extensions-builtin/Lora/scripts/lora_script.py index 60b9eb64..544b228d 100644 --- a/extensions-builtin/Lora/scripts/lora_script.py +++ b/extensions-builtin/Lora/scripts/lora_script.py @@ -3,7 +3,7 @@ import torch import lora
import extra_networks_lora
import ui_extra_networks_lora
-from modules import script_callbacks, ui_extra_networks, extra_networks
+from modules import script_callbacks, ui_extra_networks, extra_networks, shared
def unload():
@@ -28,3 +28,8 @@ torch.nn.Conv2d.forward = lora.lora_Conv2d_forward script_callbacks.on_model_loaded(lora.assign_lora_names_to_compvis_modules)
script_callbacks.on_script_unloaded(unload)
script_callbacks.on_before_ui(before_ui)
+
+
+shared.options_templates.update(shared.options_section(('extra_networks', "Extra Networks"), {
+ "lora_apply_to_outputs": shared.OptionInfo(False, "Apply Lora to outputs rather than inputs when possible (experimental)"),
+}))
|