From 1773287097eb73c8572a925be8e863c9528b5b8f Mon Sep 17 00:00:00 2001 From: xuanzic Date: Wed, 7 Aug 2024 18:32:57 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: xuanzic --- nemo/export/multimodal/build.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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