diff --git a/nemo/export/multimodal/build.py b/nemo/export/multimodal/build.py index 473ede915e94..8ee3fa1c05e7 100644 --- a/nemo/export/multimodal/build.py +++ b/nemo/export/multimodal/build.py @@ -208,7 +208,10 @@ def forward(self, images): return vision_x encoder = AutoModel.from_pretrained( - vision_config["from_pretrained"], torch_dtype=torch.bfloat16, trust_remote_code=True, attn_implementation='eager' + vision_config["from_pretrained"], + torch_dtype=torch.bfloat16, + trust_remote_code=True, + attn_implementation='eager', ) vision_encoder = encoder.vision_model hf_config = encoder.config @@ -326,7 +329,10 @@ def forward(self, images): return vision_x encoder = AutoModel.from_pretrained( - vision_config["from_pretrained"], torch_dtype=torch.bfloat16, trust_remote_code=True, attn_implementation='eager' + vision_config["from_pretrained"], + torch_dtype=torch.bfloat16, + trust_remote_code=True, + attn_implementation='eager', ) vision_encoder = encoder.vision_model hf_config = encoder.config