Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

add support for nunchaku-lite models and engine

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-04 15:30:45 +02:00
parent 9a7fa12218
commit a1df374773
11 changed files with 89 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/attention.py
+2 -2
View File
@@ -291,7 +291,7 @@ def set_diffusers_attention(pipe, quiet = False):
orig_get_kernel = None
def get_kernel_hijack(repo_id, revision=None, version=None, backend=None, user_agent=None, trust_remote_code: bool | list[str] = False): # pylint: disable=unused-argument
log.debug(f'Attention dispatcher hub: repo="{repo_id}" revision={revision} version={version} backend={backend}')
user_agent = 'kernels/0.14.1'
user_agent = 'kernels/0.16.0'
module = None
try:
module = orig_get_kernel(repo_id, revision=revision, version=version, backend=backend, user_agent=user_agent, trust_remote_code=True)
@@ -309,7 +309,7 @@ def get_hf_api_hijack(user_agent = None): # pylint: disable=unused-argument
def hijack_kernels():
global orig_get_kernel # pylint: disable=global-statement
try:
install('kernels==0.14.1')
install('kernels==0.16.0')
import kernels
import kernels.utils
log.debug(f'Attention dispatcher: kernels={kernels.__version__}')
modules/model_quant.py
+2
View File
@@ -198,6 +198,8 @@ def check_nunchaku(module: str = ''):
from modules import shared
if 'nunchaku' not in shared.opts.sd_model_checkpoint.lower():
return False
if 'nunchaku-lite' in shared.opts.sd_model_checkpoint.lower():
return False
base_path = shared.opts.sd_model_checkpoint.split('+')[0]
for v in shared.reference_models.values():
if v.get('path', '') != base_path: