From 8f0f467c2070e6068de414fe32edf3b860e95343 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Mon, 29 Jul 2024 07:22:05 -0600 Subject: [PATCH] Switch back to old ilora --- 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 d4ce39a6..3f5c2b82 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.ilora2 import InstantLoRAModule +from toolkit.models.ilora 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