From 3cdc26af3083e8fbe9763005f4cb382428bf6230 Mon Sep 17 00:00:00 2001 From: Won-Kyu Park Date: Sun, 15 Sep 2024 23:29:45 +0900 Subject: [PATCH] fix lora without backup --- extensions-builtin/Lora/networks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index 46c9ac1ff..dfdf3c7e6 100644 --- a/extensions-builtin/Lora/networks.py +++ b/extensions-builtin/Lora/networks.py @@ -397,7 +397,7 @@ def network_restore_weights_from_backup(self: Union[torch.nn.Conv2d, torch.nn.Li if weights_backup is None and bias_backup is None: return - if weights_backup is True or weights_backup == (True, True): # fake backup + if shared.opts.lora_without_backup_weight: return if weights_backup is not None: @@ -567,7 +567,7 @@ def network_apply_weights(self: Union[torch.nn.Conv2d, torch.nn.Linear, torch.nn extra_network_lora.errors[net.name] = extra_network_lora.errors.get(net.name, 0) + 1 - if weights_backup is True or weights_backup == (True, True): # fake backup + if shared.opts.lora_without_backup_weight: self.network_weights_backup = None self.network_bias_backup = None else: