Merge pull request #13178 from wpdong0727/fix-lora-bias-backup-reset

fix: lora-bias-backup don't reset cache
2023-09-30 09:48:38 +03:00 · 2023-09-30 09:48:38 +03:00 · a0e979badb
parent 3aa9f01bdc 7d4d871d46
commit a0e979badb
1 changed files with 1 additions and 0 deletions
--- a/extensions-builtin/Lora/networks.py
+++ b/extensions-builtin/Lora/networks.py
@ -418,6 +418,7 @@ def network_forward(module, input, original_forward):
 def network_reset_cached_weight(self: Union[torch.nn.Conv2d, torch.nn.Linear]):
    self.network_current_names = ()
    self.network_weights_backup = None
+    self.network_bias_backup = None


 def network_Linear_forward(self, input):