From c8597ca84e6aa72be7919d2c51d3975c52c9ca1a Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Fri, 6 Feb 2026 00:10:18 +0000 Subject: [PATCH] feat: hide Nunchaku reference models on non-CUDA backends Filter out reference entries tagged "nunchaku" from Extra Networks when the active backend is not CUDA, since Nunchaku requires NVIDIA GPUs. Entries remain in shared.reference_models for programmatic lookup but are not yielded to the UI. --- modules/ui_extra_networks_checkpoints.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/ui_extra_networks_checkpoints.py b/modules/ui_extra_networks_checkpoints.py index 160fa3327..b179d55c6 100644 --- a/modules/ui_extra_networks_checkpoints.py +++ b/modules/ui_extra_networks_checkpoints.py @@ -3,7 +3,7 @@ import html import json import concurrent from datetime import datetime -from modules import shared, ui_extra_networks, sd_models, modelstats, paths +from modules import shared, ui_extra_networks, sd_models, modelstats, paths, devices from modules.json_helpers import readfile @@ -88,6 +88,9 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage): path = f'{v.get("path", "")}' tag = v.get('tags', '') + if tag == 'nunchaku' and devices.backend != 'cuda': + count['hidden'] += 1 + continue if tag in count: count[tag] += 1 elif tag != '':