Skip to content

Commit 6552653

Browse files
committed
update
1 parent 969c3ab commit 6552653

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/mochi/pipeline_mochi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -618,7 +618,7 @@ def __call__(
618618

619619
noise_pred = self.transformer(
620620
hidden_states=latent_model_input,
621-
timestep=timestep / 1000,
621+
timestep=timestep,
622622
encoder_hidden_states=prompt_embeds,
623623
joint_attention_kwargs=self.joint_attention_kwargs,
624624
return_dict=False,

0 commit comments

Comments
 (0)