fix lint

open-mmlab · Nov 10, 2023 · f1d76cc · f1d76cc
1 parent 889fc96
commit f1d76cc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mmagic/models/editors/fastcomposer/fastcomposer_util.py b/mmagic/models/editors/fastcomposer/fastcomposer_util.py
@@ -12,10 +12,13 @@
 from transformers import (CLIPModel, CLIPPreTrainedModel, CLIPTextModel,
                           CLIPVisionConfig, CLIPVisionModel)
 from transformers.modeling_outputs import BaseModelOutputWithPooling
+
 try:
     from transformers.models.clip.modeling_clip import _expand_mask
 except:
-    from transformers.models.clip.modeling_clip import _prepare_4d_attention_mask as _expand_mask
+    from transformers.models.clip.modeling_clip import \
+        _prepare_4d_attention_mask as _expand_mask
+
 
 class FastComposerModel(nn.Module):
     """FastComposerModel is based on the StableDiffusion Model and the Clip