From 892dd456f76c83e3b8a308aad5d0b6f8b8903d90 Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Mon, 16 Jun 2025 02:43:49 +0300
Subject: [PATCH] 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'):