mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
Remove line that was intended to save memory, but wasn't actually having any effect.
This commit is contained in:
parent
4af258615f
commit
6e4de001f1
@ -544,12 +544,8 @@ class LoRAModelRaw(RawModel): # (torch.nn.Module):
|
||||
for layer_key, values in state_dict.items():
|
||||
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)
|
||||
model.layers[layer_key] = layer
|
||||
|
||||
return model
|
||||
|
||||
@staticmethod
|
||||
|
Loading…
Reference in New Issue
Block a user