From 40ddfd4499d641d80528359efd13a3fa78142d19 Mon Sep 17 00:00:00 2001 From: "yijing.wq" Date: Mon, 18 Sep 2023 21:22:56 +0800 Subject: [PATCH] fix maybe_allow_in_graph of diffusers update version Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/14077408 * fix_maybe_allow --- modelscope/models/cv/image_super_resolution_pasd/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modelscope/models/cv/image_super_resolution_pasd/attention.py b/modelscope/models/cv/image_super_resolution_pasd/attention.py index 1ec566de..825a98a9 100644 --- a/modelscope/models/cv/image_super_resolution_pasd/attention.py +++ b/modelscope/models/cv/image_super_resolution_pasd/attention.py @@ -7,7 +7,7 @@ import torch import torch.nn.functional as F from diffusers.models.attention_processor import Attention from diffusers.models.embeddings import CombinedTimestepLabelEmbeddings -from diffusers.utils import maybe_allow_in_graph +from diffusers.utils.torch_utils import maybe_allow_in_graph from torch import nn