Skip to content

Commit 7cfcce0

Browse files
committed
debug added for decoded video
1 parent 3e6b745 commit 7cfcce0

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p1.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -263,6 +263,11 @@ def __call__(
263263
if output_type == "latent":
264264
return jnp.transpose(latents_denorm_bcthw, (0, 2, 3, 4, 1))
265265
decoded_video = self._decode_latents_to_video(latents_denorm_bcthw)
266+
jax.debug.print("Decoded video: min={mn}, max={mx}, mean={mean}, std={std}",
267+
mn=jnp.min(decoded_video),
268+
mx=jnp.max(decoded_video),
269+
mean=jnp.mean(decoded_video),
270+
std=jnp.std(decoded_video))
266271
return decoded_video
267272

268273

0 commit comments

Comments
 (0)