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

import sdnq before load

Browse Source
This commit is contained in:
Disty0
2025-10-25 20:06:13 +03:00
parent f6603ab549
commit 975361b404
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pipelines/generic.py
+4
View File
@@ -12,6 +12,8 @@ def load_transformer(repo_id, cls_name, load_config={}, subfolder="transformer",
transformer = None
jobid = shared.state.begin('Load DiT')
try:
if 'sdnq-' not in repo_id.lower():
from modules import sdnq # register to diffusers and transformers
load_args, quant_args = model_quant.get_dit_args(load_config, module='Model', device_map=True, allow_quant=allow_quant, modules_to_not_convert=modules_to_not_convert, modules_dtype_dict=modules_dtype_dict)
quant_type = model_quant.get_quant_type(quant_args)
dtype = dtype or devices.dtype
@@ -83,6 +85,8 @@ def load_text_encoder(repo_id, cls_name, load_config={}, subfolder="text_encoder
text_encoder = None
jobid = shared.state.begin('Load TE')
try:
if 'sdnq-' not in repo_id.lower():
from modules import sdnq # register to diffusers and transformers
load_args, quant_args = model_quant.get_dit_args(load_config, module='TE', device_map=True, allow_quant=allow_quant, modules_to_not_convert=modules_to_not_convert, modules_dtype_dict=modules_dtype_dict)
quant_type = model_quant.get_quant_type(quant_args)
dtype = dtype or devices.dtype