Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "ae0fb243bcc86b7f737e5d8898137c8aaf2ee0b6"
Commit e3d0a9a4 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix potential issue with text projection matrix multiplication.

parent cc44ade7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/sd1_clip.py
View file @ e3d0a9a4
...@@ -156,7 +156,7 @@ class SD1ClipModel(torch.nn.Module, ClipTokenWeightEncoder): ...@@ -156,7 +156,7 @@ class SD1ClipModel(torch.nn.Module, ClipTokenWeightEncoder):
pooled_output = outputs.pooler_output pooled_output = outputs.pooler_output
if self.text_projection is not None: if self.text_projection is not None:
pooled_output = pooled_output.to(self.text_projection.device) @ self.text_projection pooled_output = pooled_output.float().to(self.text_projection.device) @ self.text_projection.float()
return z.float(), pooled_output.float() return z.float(), pooled_output.float()
def encode(self, tokens): def encode(self, tokens):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment