From e2f656dc3ceab99f6cdbe4cab9eb37a0eff711e7 Mon Sep 17 00:00:00 2001
From: xiaokun-hadoop <78018550+xiaokun-hadoop@users.noreply.github.com>
Date: Thu, 6 Feb 2025 18:38:45 +0800
Subject: [PATCH] Update sdxl_prompt2prompt_mapper.py

Standardize the use of attention mechanism related modules
---
 data_juicer/ops/mapper/sdxl_prompt2prompt_mapper.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/data_juicer/ops/mapper/sdxl_prompt2prompt_mapper.py b/data_juicer/ops/mapper/sdxl_prompt2prompt_mapper.py
index b8d30b6bd..c268d6f54 100644
--- a/data_juicer/ops/mapper/sdxl_prompt2prompt_mapper.py
+++ b/data_juicer/ops/mapper/sdxl_prompt2prompt_mapper.py
@@ -8,6 +8,8 @@
 from data_juicer.ops.op_fusion import LOADED_IMAGES
 from data_juicer.utils.lazy_loader import LazyLoader
 from data_juicer.utils.model_utils import get_model, prepare_model
+from diffusers.models.attention import Attention
+
 
 diffusers = LazyLoader('diffusers', 'diffusers')
 torch = LazyLoader('torch', 'torch')