Co-authored-by: graemeniedermayer graemeniedermayer@users.noreply.github.com
This commit is contained in:
DenOfEquity
2025-02-27 17:54:44 +00:00
committed by GitHub
parent 8dd92501e6
commit f23bc80d2f
6 changed files with 1184 additions and 22 deletions

View File

@@ -141,7 +141,7 @@ class ClassicTextProcessingEngine:
if self.return_pooled:
pooled_output = outputs.pooler_output
if self.text_projection:
if self.text_projection and self.embedding_key != 'clip_l':
pooled_output = self.text_encoder.transformer.text_projection(pooled_output)
z.pooled = pooled_output