Update flux/transformer_flux_simple.py

flux/transformer_flux_simple.py

@@ -22,7 +22,7 @@ import torch.nn.functional as F
 from diffusers.configuration_utils import ConfigMixin, register_to_config
 from .lora.peft import PeftAdapterMixin
 from diffusers.models.attention import FeedForward
-from .attention_processor import Attention, FluxAttnProcessor2_0, FluxSingleAttnProcessor2_0
+from .attention_processor_og import Attention, FluxAttnProcessor2_0, FluxSingleAttnProcessor2_0
 from diffusers.models.modeling_utils import ModelMixin
 from .normalization import AdaLayerNormContinuous, AdaLayerNormZero, AdaLayerNormZeroSingle
 from diffusers.utils import USE_PEFT_BACKEND, is_torch_version, logging, scale_lora_layers, unscale_lora_layers