try import expand mask from transformers

open-mmlab · Nov 10, 2023 · 889fc96 · 889fc96
1 parent 991d535
commit 889fc96
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mmagic/models/editors/fastcomposer/fastcomposer_util.py b/mmagic/models/editors/fastcomposer/fastcomposer_util.py
@@ -12,8 +12,10 @@
 from transformers import (CLIPModel, CLIPPreTrainedModel, CLIPTextModel,
                           CLIPVisionConfig, CLIPVisionModel)
 from transformers.modeling_outputs import BaseModelOutputWithPooling
-from transformers.models.clip.modeling_clip import _expand_mask
-
+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
 
 class FastComposerModel(nn.Module):
     """FastComposerModel is based on the StableDiffusion Model and the Clip