OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-17 11:50:18 +08:00
Code Issues Packages Projects Releases 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
Show Stats 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) setattr(obj, field, None)
return 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): 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):