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

k1nos · Sep 10, 2023 · 7d4d871 · 7d4d871
1 parent 9246423
commit 7d4d871
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):