Skip to content

Video training part 2 #218

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 15 commits into from
Aug 5, 2025
Merged

Video training part 2 #218

merged 15 commits into from
Aug 5, 2025

Conversation

entrpn
Copy link
Collaborator

@entrpn entrpn commented Jul 31, 2025 •
edited
Loading

  • Use fractional per_device_batch_size
  • Shard projections for better memory allocation.
  • Remote jitted splash kernel to reduce memory. No performance implication.

@github-actions GitHub Actions
Copy link

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

@github-actions GitHub Actions
Copy link

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

@github-actions GitHub Actions
Copy link

github-actions bot commented Aug 1, 2025

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

@entrpn entrpn requested a review from coolkp August 1, 2025 15:28
@github-actions GitHub Actions
Copy link

github-actions bot commented Aug 1, 2025

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

@github-actions GitHub Actions
Copy link

github-actions bot commented Aug 5, 2025

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

@entrpn entrpn merged commit fc46fcc into main Aug 5, 2025
2 of 4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@coolkp coolkp coolkp approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@entrpn @coolkp