diff --git a/models.py b/models.py
index aefe1ee..02f29d8 100644
--- a/models.py
+++ b/models.py
@@ -12,9 +12,9 @@
 )
 
 from src.mask_ip_controller import *
-from src.ip_adapter.attention_processor import AttnProcessor2_0 as AttnProcessor
-from src.ip_adapter.attention_processor import IPAttnProcessor2_0 as IPAttnProcessor
-from src.ip_adapter.mask_attention_processor import IPAttnProcessor2_0WithIPMaskController
+from src.attention_processor import AttnProcessor2_0 as AttnProcessor
+from src.attention_processor import IPAttnProcessor2_0 as IPAttnProcessor
+from src.mask_attention_processor import IPAttnProcessor2_0WithIPMaskController
 
 def tokenize_captions(tokenizer, captions):
     inputs = tokenizer(