Always return unprojected pooled output for gligen.

This commit is contained in:
comfyanonymous
2024-02-25 07:20:31 -05:00
parent 1cb3f6a83b
commit c2cb8e889b
6 changed files with 38 additions and 28 deletions

View File

@@ -133,7 +133,7 @@ class CLIPTextModel(torch.nn.Module):
def forward(self, *args, **kwargs):
x = self.text_model(*args, **kwargs)
out = self.text_projection(x[2])
return (x[0], x[1], out)
return (x[0], x[1], out, x[2])
class CLIPVisionEmbeddings(torch.nn.Module):