Skip to content

Use jnp.tile instead of deprecated pltpu.repeat#401

Merged
copybara-service[bot] merged 1 commit into
mainfrom
jax_update
May 11, 2026
Merged

Use jnp.tile instead of deprecated pltpu.repeat#401
copybara-service[bot] merged 1 commit into
mainfrom
jax_update

Conversation

@eltsai
Copy link
Copy Markdown
Collaborator

@eltsai eltsai commented May 10, 2026

Creating this PR based on CL.

I ran the custom kernel + ulysses script:

==================================================
  TIMING SUMMARY
==================================================
  Load (checkpoint):     911.3s
  Compile:               238.6s
  Inference:             138.4s
  ────────────────────────────────────────
  Conditioning:            2.4s
  Denoise Total:         131.1s
  VAE Decode:              4.8s
==================================================

The video quality looks good.

@eltsai eltsai requested a review from entrpn as a code owner May 10, 2026 05:48
@github-actions
Copy link
Copy Markdown

@eltsai eltsai self-assigned this May 10, 2026
Copy link
Copy Markdown
Collaborator

@Perseus14 Perseus14 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

WANVace tests failed - this is unrelated to the current PR.

@copybara-service copybara-service Bot merged commit afb1e74 into main May 11, 2026
17 of 18 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants