From 889fc966807dbc916eadcee851802e7bc795f525 Mon Sep 17 00:00:00 2001 From: zengyh1900 Date: Fri, 10 Nov 2023 13:19:59 +0800 Subject: [PATCH] try import expand mask from transformers --- mmagic/models/editors/fastcomposer/fastcomposer_util.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mmagic/models/editors/fastcomposer/fastcomposer_util.py b/mmagic/models/editors/fastcomposer/fastcomposer_util.py index fa9bf9c56a..e6bf053f93 100644 --- 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