diff --git a/nemo/collections/multimodal/modules/stable_diffusion/encoders/modules.py b/nemo/collections/multimodal/modules/stable_diffusion/encoders/modules.py index ab33532c3c1fa..0443d75a61e83 100644 --- a/nemo/collections/multimodal/modules/stable_diffusion/encoders/modules.py +++ b/nemo/collections/multimodal/modules/stable_diffusion/encoders/modules.py @@ -689,6 +689,8 @@ def load_model(self, cfg, state_dict): model_cfg=cfg, model_parallel_config=ModelParallelConfig(), padded_vocab_size=padded_vocab_size, + vision_transformer_config=None, # assumed mcore to be false + text_transformer_config=None, pre_process=cfg.text.pre_process, post_process=cfg.text.post_process, )