ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-28 18:21:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Varous bug fixes

Browse Source
This commit is contained in:
Jaret Burkett
2024-07-10 15:20:04 -06:00
parent c008405480
commit c062b7716c
9 changed files with 294 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/models/te_adapter.py
View File

@@ -246,7 +246,7 @@ class TEAdapter(torch.nn.Module):
if self.adapter_ref().config.text_encoder_arch == "t5":
self.token_size = self.te_ref().config.d_model
else:
self.token_size = self.te_ref().config.hidden_size
self.token_size = self.te_ref().config.target_hidden_size
# add text projection if is sdxl
self.text_projection = None