AUTOMATIC1111/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-04-29 10:41:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

RAM optimization round 2

Browse Source
This commit is contained in:
AUTOMATIC1111
2023-08-16 09:55:35 +03:00
parent 85fcb7b8df
commit 86221269f9
2 changed files with 48 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/Lora/networks.py
View File

@@ -304,7 +304,10 @@ def network_apply_weights(self: Union[torch.nn.Conv2d, torch.nn.Linear, torch.nn
wanted_names = tuple((x.name, x.te_multiplier, x.unet_multiplier, x.dyn_dim) for x in loaded_networks)
weights_backup = getattr(self, "network_weights_backup", None)
if weights_backup is None:
if weights_backup is None and wanted_names != ():
if current_names != ():
raise RuntimeError("no backup weights found and current weights are not unchanged")
if isinstance(self, torch.nn.MultiheadAttention):
weights_backup = (self.in_proj_weight.to(devices.cpu, copy=True), self.out_proj.weight.to(devices.cpu, copy=True))
else: