diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-28 17:04:35 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-28 17:04:35 +0000 |
| commit | 2abd89acc66419abf2eee9b03fd093f2737670de (patch) | |
| tree | f8cb980542275899882586f7457bcfc7f4b2234b /extensions-builtin/Lora/scripts/lora_script.py | |
| parent | 91c8d0dcfc9a5d46aec47fd3eb34c633c104b5bb (diff) | |
| download | stable-diffusion-webui-gfx803-2abd89acc66419abf2eee9b03fd093f2737670de.tar.gz stable-diffusion-webui-gfx803-2abd89acc66419abf2eee9b03fd093f2737670de.tar.bz2 stable-diffusion-webui-gfx803-2abd89acc66419abf2eee9b03fd093f2737670de.zip | |
index on master: 91c8d0d Merge pull request #7231 from EllangoK/master
Diffstat (limited to 'extensions-builtin/Lora/scripts/lora_script.py')
| -rw-r--r-- | extensions-builtin/Lora/scripts/lora_script.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/scripts/lora_script.py b/extensions-builtin/Lora/scripts/lora_script.py index 2e860160..a385ae94 100644 --- a/extensions-builtin/Lora/scripts/lora_script.py +++ b/extensions-builtin/Lora/scripts/lora_script.py @@ -10,6 +10,7 @@ from modules import script_callbacks, ui_extra_networks, extra_networks, shared def unload():
torch.nn.Linear.forward = torch.nn.Linear_forward_before_lora
torch.nn.Conv2d.forward = torch.nn.Conv2d_forward_before_lora
+ torch.nn.modules.linear.NonDynamicallyQuantizableLinear.forward = torch.nn.NonDynamicallyQuantizableLinear_forward_before_lora
def before_ui():
@@ -23,8 +24,12 @@ if not hasattr(torch.nn, 'Linear_forward_before_lora'): if not hasattr(torch.nn, 'Conv2d_forward_before_lora'):
torch.nn.Conv2d_forward_before_lora = torch.nn.Conv2d.forward
+if not hasattr(torch.nn, 'NonDynamicallyQuantizableLinear_forward_before_lora'):
+ torch.nn.NonDynamicallyQuantizableLinear_forward_before_lora = torch.nn.modules.linear.NonDynamicallyQuantizableLinear.forward
+
torch.nn.Linear.forward = lora.lora_Linear_forward
torch.nn.Conv2d.forward = lora.lora_Conv2d_forward
+torch.nn.modules.linear.NonDynamicallyQuantizableLinear.forward = lora.lora_NonDynamicallyQuantizableLinear_forward
script_callbacks.on_model_loaded(lora.assign_lora_names_to_compvis_modules)
script_callbacks.on_script_unloaded(unload)
|