Skip to content

Commit 887f079

Browse files
committed
removing diffusion steps time benchmark logs to see if it speedsup things
1 parent e8f964e commit 887f079

1 file changed

Lines changed: 1 addition & 7 deletions

File tree

src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1237,9 +1237,7 @@ def run_connectors(graphdef, state, hidden_states, attention_mask):
12371237
)
12381238

12391239
import time
1240-
total_diffusion_time = 0.0
12411240
for i, t in enumerate(timesteps):
1242-
step_start_time = time.perf_counter()
12431241

12441242
# Isolate input sharding to scan_layers=False to avoid affecting the standard path
12451243
latents_jax_sharded = latents_jax
@@ -1299,11 +1297,7 @@ def run_connectors(graphdef, state, hidden_states, attention_mask):
12991297
latents_jax = latents_step
13001298
audio_latents_jax = audio_latents_step
13011299

1302-
latents_jax.block_until_ready()
1303-
step_duration = time.perf_counter() - step_start_time
1304-
total_diffusion_time += step_duration
1305-
max_logging.log(f"[Tuning] Diffusion Step {i} e2e time: {step_duration:.4f} seconds")
1306-
max_logging.log(f"[Tuning] Total pure diffusion time (all steps): {total_diffusion_time:.4f} seconds")
1300+
13071301

13081302
# 8. Decode Latents
13091303
if guidance_scale > 1.0:

0 commit comments

Comments
 (0)