OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-01-19 21:00:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

use assign=True for some cases

Browse Source
This commit is contained in:
Won-Kyu Park 2024-09-18 02:18:00 +09:00
parent eee7294200
commit 6675d1f090
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_disable_initialization.py
View File

@ -176,6 +176,11 @@ class LoadStateDictOnMeta(ReplaceHelper):
def load_from_state_dict(original, module, state_dict, prefix, *args, **kwargs):
used_param_keys = []
if type(module) in (torch.nn.Linear, torch.nn.Conv2d, torch.nn.GroupNorm, torch.nn.LayerNorm,):
# HACK add assign=True to local_metadata for some cases
args[0]['assign_to_params_buffers'] = True
for name, param in module._parameters.items():
if param is None:
continue