From 0cd598958d6c88371d136c519aa970427df29f51 Mon Sep 17 00:00:00 2001 From: hahminlew Date: Tue, 23 Sep 2025 17:58:15 +0900 Subject: [PATCH] fix bug in import attention_processor --- models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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(