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

Merge 62da1aaf280f9094ad20361299d34293c36aa66a into 374bb6cc384d2a19422c0b07d69de0a41d1f3f4d

Browse Source
This commit is contained in:
Won-Kyu Park 2025-03-06 23:16:52 +08:00 committed by GitHub
commit d91dab05d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_disable_initialization.py
View File

@ -188,7 +188,7 @@ class LoadStateDictOnMeta(ReplaceHelper):
if param.is_meta: if param.is_meta:
dtype = sd_param.dtype if sd_param is not None else param.dtype dtype = sd_param.dtype if sd_param is not None else param.dtype
module._parameters[name] = torch.nn.parameter.Parameter(torch.zeros_like(param, device=device, dtype=dtype), requires_grad=param.requires_grad) module._parameters[name] = torch.nn.parameter.Parameter(torch.empty_like(param, device=device, dtype=dtype), requires_grad=param.requires_grad)
for name in module._buffers: for name in module._buffers:
key = prefix + name key = prefix + name