diff --git a/modules/model_pixart.py b/modules/model_pixart.py index 4ae7eb21c..d57079e4e 100644 --- a/modules/model_pixart.py +++ b/modules/model_pixart.py @@ -19,6 +19,7 @@ def load_pixart(checkpoint_info, diffusers_load_config={}): kwargs = { 'transformer': transformer } if t5 is not None: kwargs['text_encoder'] = t5 + diffusers_load_config.pop('variant', None) pipe = diffusers.PixArtSigmaPipeline.from_pretrained( 'PixArt-alpha/PixArt-Sigma-XL-2-1024-MS', cache_dir = shared.opts.diffusers_dir,