convert to python v3.9

6a147db1 · camenduru · GitHub · 9d7390d2 · 6a147db1
Commit 6a147db1 authored Mar 27, 2023 by camenduru Committed by GitHub Mar 27, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

extensions-builtin/Lora/lora.py extensions-builtin/Lora/lora.py +1 -1

No files found.
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -296,7 +296,7 @@ def lora_apply_weights(self: Union[torch.nn.Conv2d, torch.nn.Linear, torch.nn.Mu
        setattr(self, "lora_current_names", wanted_names)


-def lora_reset_cached_weight(self: torch.nn.Conv2d | torch.nn.Linear):
+def lora_reset_cached_weight(self: Union[torch.nn.Conv2d, torch.nn.Linear]):
    setattr(self, "lora_current_names", ())
    setattr(self, "lora_weights_backup", None)