fix: lora-bias-backup don't reset cache

2023-09-10 17:53:42 +08:00 · 2023-09-10 17:53:42 +08:00 · 7d4d871d46
parent 924642331b
commit 7d4d871d46
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):