diff --git a/modules/sd_models_utils.py b/modules/sd_models_utils.py index 4430527e5..0d201779c 100644 --- a/modules/sd_models_utils.py +++ b/modules/sd_models_utils.py @@ -162,6 +162,10 @@ def apply_function_to_model(sd_model, function, options, op=None): sd_model.unet = function(sd_model.unet, op="unet", sd_model=sd_model) if hasattr(sd_model, 'transformer') and hasattr(sd_model.transformer, 'config'): sd_model.transformer = function(sd_model.transformer, op="transformer", sd_model=sd_model) + if hasattr(sd_model, 'transformer_2') and hasattr(sd_model.transformer_2, 'config'): + sd_model.transformer_2 = function(sd_model.transformer_2, op="transformer_2", sd_model=sd_model) + if hasattr(sd_model, 'transformer_3') and hasattr(sd_model.transformer_3, 'config'): + sd_model.transformer_3 = function(sd_model.transformer_3, op="transformer_3", sd_model=sd_model) if hasattr(sd_model, 'decoder_pipe') and hasattr(sd_model, 'decoder'): sd_model.decoder = None sd_model.decoder = sd_model.decoder_pipe.decoder = function(sd_model.decoder_pipe.decoder, op="decoder_pipe.decoder", sd_model=sd_model) diff --git a/pipelines/model_wanai.py b/pipelines/model_wanai.py index 246d10123..e7cc61107 100644 --- a/pipelines/model_wanai.py +++ b/pipelines/model_wanai.py @@ -24,6 +24,7 @@ def load_transformer(repo_id, diffusers_load_config={}, subfolder='transformer') fn, cache_dir=shared.opts.hfcache_dir, **load_args, + **quant_args, ) else: shared.log.debug(f'Load model: type=WanAI {subfolder}="{repo_id}" quant="{model_quant.get_quant_type(quant_args)}" args={load_args}')