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

fix to support float8_*

Browse Source
This commit is contained in:
Won-Kyu Park 2024-08-31 23:22:43 +09:00
parent 219a0e2429
commit 9e57c722b2
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/sd_disable_initialization.py
View File

@ -160,7 +160,7 @@ class LoadStateDictOnMeta(ReplaceHelper):
self.state_dict = state_dict
self.device = device
self.weight_dtype_conversion = weight_dtype_conversion or {}
self.default_dtype = self.weight_dtype_conversion.get('')
self.default_dtype = self.weight_dtype_conversion.get('', None)
def get_weight_dtype(self, key):
key_first_term, _ = key.split('.', 1)
@ -183,7 +183,11 @@ class LoadStateDictOnMeta(ReplaceHelper):
key = prefix + name
sd_param = sd.pop(key, None)
if sd_param is not None:
state_dict[key] = sd_param.to(dtype=self.get_weight_dtype(key))
dtype = self.get_weight_dtype(key)
if dtype is None:
state_dict[key] = sd_param
else:
state_dict[key] = sd_param.to(dtype=dtype)
used_param_keys.append(key)
if param.is_meta: