From 892dd456f76c83e3b8a308aad5d0b6f8b8903d90 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 16 Jun 2025 02:43:49 +0300 Subject: [PATCH 1/2] Fix Nunchaku --- CHANGELOG.md | 3 ++- modules/model_flux.py | 8 ++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 51a1c7ac7..e9ce65586 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,6 +1,6 @@ # Change Log for SD.Next -## Update for 2025-06-15 +## Update for 2025-06-16 - **Feature** - Support for Python 3.13 @@ -42,6 +42,7 @@ - TAESD previews with PixArt and Lumina 2 - VAE Tiling with non-default tile sizes - Lumina 2 with IPEX + - Nunchaku updated repo ## Update for 2025-06-02 diff --git a/modules/model_flux.py b/modules/model_flux.py index dab00d86a..b5f482f2b 100644 --- a/modules/model_flux.py +++ b/modules/model_flux.py @@ -115,11 +115,11 @@ def load_quants(kwargs, repo_id, cache_dir, allow_quant): nunchaku_precision = nunchaku.utils.get_precision() nunchaku_repo = None if 'dev' in repo_id: - nunchaku_repo = f"mit-han-lab/svdq-{nunchaku_precision}-flux.1-dev" + nunchaku_repo = f"mit-han-lab/nunchaku-flux.1-dev/svdq-{nunchaku_precision}_r32-flux.1-dev.safetensors" elif 'schnell' in repo_id: - nunchaku_repo = f"mit-han-lab/svdq-{nunchaku_precision}-flux.1-schnell" + nunchaku_repo = f"mit-han-lab/nunchaku-flux.1-schnell/svdq-{nunchaku_precision}_r32-flux.1-schnell.safetensors" elif 'shuttle' in repo_id: - nunchaku_repo = 'mit-han-lab/svdq-fp4-shuttle-jaguar' + nunchaku_repo = f"mit-han-lab/nunchaku-shuttle-jaguar/svdq-{nunchaku_precision}_r32-shuttle-jaguar.safetensors" else: shared.log.error(f'Load module: quant=Nunchaku module=transformer repo="{repo_id}" unsupported') if nunchaku_repo is not None: @@ -135,7 +135,7 @@ def load_quants(kwargs, repo_id, cache_dir, allow_quant): if 'text_encoder_2' not in kwargs and model_quant.check_nunchaku('TE'): import nunchaku nunchaku_precision = nunchaku.utils.get_precision() - nunchaku_repo = 'mit-han-lab/svdq-flux.1-t5' + nunchaku_repo = 'mit-han-lab/nunchaku-t5/awq-int4-flux.1-t5xxl.safetensors' shared.log.debug(f'Load module: quant=Nunchaku module=t5 repo="{nunchaku_repo}" precision={nunchaku_precision}') kwargs['text_encoder_2'] = nunchaku.NunchakuT5EncoderModel.from_pretrained(nunchaku_repo, torch_dtype=devices.dtype) elif 'text_encoder_2' not in kwargs and model_quant.check_quant('TE'): From c4f98a65a41d7538582de71ec5b2bd8f435ff9a8 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 16 Jun 2025 02:46:48 +0300 Subject: [PATCH 2/2] Update wiki --- wiki | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wiki b/wiki index fe2da3ae2..2ca67daca 160000 --- a/wiki +++ b/wiki @@ -1 +1 @@ -Subproject commit fe2da3ae29008d10ac1ef322752d881fa14a2e7e +Subproject commit 2ca67dacaccb7ef4c0595b19407fc5a93167008b