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