diff --git a/modules/mit_nunchaku.py b/modules/mit_nunchaku.py index ea616fe5b..c15ccbd62 100644 --- a/modules/mit_nunchaku.py +++ b/modules/mit_nunchaku.py @@ -65,12 +65,16 @@ def install_nunchaku(force=False): log.error(f'Nunchaku: torch={torch.__version__} cuda="unknown"') return False if cuda_ver.startswith('cu13'): - nunchaku_versions = ['1.2.1', '1.2.0', '1.1.0', '1.0.2', '1.0.1'] + nunchaku_versions = ['1.3.0dev20260306', '1.2.1', '1.2.0', '1.1.0', '1.0.2', '1.0.1'] else: nunchaku_versions = ['1.2.1', '1.0.2', '1.0.1'] # 1.2.0 and 1.1.0 imply cu13 but do not specify it for v in nunchaku_versions: url = f'https://github.com/nunchaku-ai/nunchaku/releases/download/v{v}/' - fn = f'nunchaku-{v}+{cuda_ver}torch{torch_ver}-cp{python_ver}-cp{python_ver}-{suffix}.whl' + if 'dev' in v: + v = v.replace('dev', '.dev') + fn = f'nunchaku-{v}+{cuda_ver}torch{torch_ver}-cp{python_ver}-cp{python_ver}-{suffix}.whl' + else: + fn = f'nunchaku-{v}+{cuda_ver}torch{torch_ver}-cp{python_ver}-cp{python_ver}-{suffix}.whl' result, _output = pip(f'install --upgrade {url+fn}', uv=False, ignore=True, quiet=True) if (result is None) or (_output == 'offline'): log.error(f'Nunchaku: install url="{url+fn}" offline mode') diff --git a/modules/model_quant.py b/modules/model_quant.py index 0cb6c05b2..5b0dcfa81 100644 --- a/modules/model_quant.py +++ b/modules/model_quant.py @@ -206,6 +206,8 @@ def check_nunchaku(module: str = ''): if module in nunchaku_modules: from modules import mit_nunchaku mit_nunchaku.install_nunchaku() + import torch._dynamo + torch._dynamo.config.recompile_limit = 16 # Set a higher limit # pylint: disable=protected-access return mit_nunchaku.ok return False diff --git a/modules/ui_definitions.py b/modules/ui_definitions.py index 82e53e406..af990c0a3 100644 --- a/modules/ui_definitions.py +++ b/modules/ui_definitions.py @@ -592,7 +592,6 @@ def create_settings(cmd_opts): "postprocessing_sep_seedvr": OptionInfo("