mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 05:57:20 +00:00
Fix some things breaking when embeddings fail to apply.
This commit is contained in:
parent
0124be4d93
commit
dfa36e6855
@ -228,6 +228,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder):
|
|||||||
if pad_extra > 0:
|
if pad_extra > 0:
|
||||||
padd_embed = self.transformer.get_input_embeddings()(torch.tensor([[self.special_tokens["pad"]] * pad_extra], device=device, dtype=torch.long), out_dtype=torch.float32)
|
padd_embed = self.transformer.get_input_embeddings()(torch.tensor([[self.special_tokens["pad"]] * pad_extra], device=device, dtype=torch.long), out_dtype=torch.float32)
|
||||||
tokens_embed = torch.cat([tokens_embed, padd_embed], dim=1)
|
tokens_embed = torch.cat([tokens_embed, padd_embed], dim=1)
|
||||||
|
attention_mask = attention_mask + [0] * pad_extra
|
||||||
|
|
||||||
embeds_out.append(tokens_embed)
|
embeds_out.append(tokens_embed)
|
||||||
attention_masks.append(attention_mask)
|
attention_masks.append(attention_mask)
|
||||||
|
Loading…
Reference in New Issue
Block a user