Unverified Commit 6a147db1 authored by camenduru's avatar camenduru Committed by GitHub
Browse files

convert to python v3.9

parent 9d7390d2
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -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)