From cb3c1fc535482003156e1c00ffe9802eeea92e97 Mon Sep 17 00:00:00 2001 From: Haoming Date: Thu, 12 Feb 2026 10:04:13 +0800 Subject: [PATCH] nunchaku --- backend/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/loader.py b/backend/loader.py index 196aced3..5307e232 100644 --- a/backend/loader.py +++ b/backend/loader.py @@ -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