Skip to content

Commit db5d401

Browse files
committed
add numpy()
1 parent 5255fee commit db5d401

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ppdiffusers/ppdiffusers/pipelines/stable_diffusion/pipeline_fastdeploy_cycle_diffusion.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -638,7 +638,9 @@ def __call__(
638638
# share_with_raw_ptr=True,
639639
# )
640640
concat_noise_pred = self.unet(
641-
sample=concat_latent_model_input, timestep=t, encoder_hidden_states=concat_prompt_embeds
641+
sample=concat_latent_model_input.numpy(),
642+
timestep=t.numpy(),
643+
encoder_hidden_states=concat_prompt_embeds.numpy(),
642644
)[0]
643645
concat_noise_pred = paddle.to_tensor(concat_noise_pred)
644646

0 commit comments

Comments
 (0)