CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

nunchaku

Browse Source
This commit is contained in:
Haoming 2026-02-12 10:04:13 +08:00
parent c6b2f28f24
commit cb3c1fc535
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/loader.py
View File

@ -307,7 +307,7 @@ def load_huggingface_component(guess, component_name, lib_name, cls_name, repo_p
_dtype_overwrite = backend.args.dynamic_args["forge_unet_storage_dtype"]
if guess.nunchaku:
storage_dtype = state_dict_dtype
storage_dtype = torch.bfloat16
logger.info(f"Using Nunchaku Model Data Type: {storage_dtype}")
elif state_dict_dtype in [torch.float8_e4m3fn, torch.float8_e5m2, "nf4", "fp4", "gguf"]:
storage_dtype = state_dict_dtype