diff options
| author | camenduru <54370274+camenduru@users.noreply.github.com> | 2023-03-27 04:40:31 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-27 04:40:31 +0300 |
| commit | 6a147db1287fe660e1bfb2ebf5b3fadc14835c69 (patch) | |
| tree | b8b8844a0a244ba6fb1ed03694fe4798e3922e44 /extensions-builtin/Lora/lora.py | |
| parent | 9d7390d2d19a8baf04ee4ebe598b96ac6ba7f97e (diff) | |
convert to python v3.9
Diffstat (limited to 'extensions-builtin/Lora/lora.py')
| -rw-r--r-- | extensions-builtin/Lora/lora.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py index 79d11e0e..696be8ea 100644 --- 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)
|