Skip to content

Commit 402ff1c

Browse files
Fix issue with ernie image. (#13393)
1 parent acd7185 commit 402ff1c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/ldm/ernie/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,7 @@ def forward(self, x, timesteps, context, **kwargs):
279279
rotary_pos_emb = self.pos_embed(torch.cat([image_ids, text_ids], dim=1)).to(x.dtype)
280280
del image_ids, text_ids
281281

282-
sample = self.time_proj(timesteps.to(dtype)).to(self.time_embedding.linear_1.weight.dtype)
282+
sample = self.time_proj(timesteps).to(dtype)
283283
c = self.time_embedding(sample)
284284

285285
shift_msa, scale_msa, gate_msa, shift_mlp, scale_mlp, gate_mlp = [

0 commit comments

Comments
 (0)