From 4c249cf607e6cb96012a51a0ed23ad1e34e7cfac Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 20 Jul 2024 16:40:57 -0600 Subject: [PATCH] Added ilora2 --- toolkit/custom_adapter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/custom_adapter.py b/toolkit/custom_adapter.py index 27120572..4ac72b6e 100644 --- a/toolkit/custom_adapter.py +++ b/toolkit/custom_adapter.py @@ -8,7 +8,7 @@ from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection, T5En from toolkit.models.clip_fusion import CLIPFusionModule from toolkit.models.clip_pre_processor import CLIPImagePreProcessor -from toolkit.models.ilora import InstantLoRAModule +from toolkit.models.ilora2 import InstantLoRAModule from toolkit.models.single_value_adapter import SingleValueAdapter from toolkit.models.te_adapter import TEAdapter from toolkit.models.te_aug_adapter import TEAugAdapter