sd3.5 integration

This commit is contained in:
grae
2024-10-25 18:39:45 -06:00
parent d4d8ad406e
commit 1363999fb1
32 changed files with 332140 additions and 11 deletions

View File

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