mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 04:11:15 -08:00
Code Issues Projects Releases Packages Wiki Activity

restore org_dtype != compute dtype case

Browse source
This commit is contained in:
Won-Kyu Park 2024-10-04 00:08:44 +09:00
parent b783a967c0
commit 310d0e6938
No known key found for this signature in database
GPG key ID: 53AA79C8C9535D15
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
extensions-builtin/Lora/networks.py
View file

@ -391,7 +391,8 @@ def restore_weights_backup(obj, field, weight):
setattr(obj, field, None)
return
getattr(obj, field).copy_(weight)
old_weight = getattr(obj, field)
old_weight.copy_(weight.to(dtype=old_weight.dtype))
def network_restore_weights_from_backup(self: Union[torch.nn.Conv2d, torch.nn.Linear, torch.nn.GroupNorm, torch.nn.LayerNorm, torch.nn.MultiheadAttention], cleanup=False):