Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove line that was intended to save memory, but wasn't actually having any effect.

Browse Source
This commit is contained in:
Ryan Dick 2024-04-04 11:29:32 -04:00
parent 4af258615f
commit 6e4de001f1
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/lora.py
View File

@ -544,12 +544,8 @@ class LoRAModelRaw(RawModel): # (torch.nn.Module):
for layer_key, values in state_dict.items(): for layer_key, values in state_dict.items():
layer = layer_cls(layer_key, values) layer = layer_cls(layer_key, values)
# lower memory consumption by removing already parsed layer values
state_dict[layer_key].clear()
layer.to(device=device, dtype=dtype) layer.to(device=device, dtype=dtype)
model.layers[layer_key] = layer model.layers[layer_key] = layer
return model return model
@staticmethod @staticmethod