From 76f225a467bcc501d214729cbf9bfa464a861d9c Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 6 Jul 2024 19:09:58 +0000 Subject: [PATCH] Fixed issue with TE adapter caption projection --- toolkit/models/te_adapter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/models/te_adapter.py b/toolkit/models/te_adapter.py index 7d147585..6da63a8d 100644 --- a/toolkit/models/te_adapter.py +++ b/toolkit/models/te_adapter.py @@ -31,7 +31,7 @@ class TEAdapterCaptionProjection(nn.Module): in_features = caption_channels self.adapter_ref: weakref.ref = weakref.ref(adapter) sd = adapter.sd_ref() - self.parent_module_ref = weakref.ref(sd.transformer.caption_projection) + self.parent_module_ref = weakref.ref(sd.unet.caption_projection) parent_module = self.parent_module_ref() self.linear_1 = nn.Linear( in_features=in_features,