Skip to content

Fix Diffusion Policy EMA with current Diffusers - #302

Open
ruziniuuuuu wants to merge 1 commit into
ARISE-Initiative:masterfrom
ruziniuuuuu:fix/diffusers-ema
Open

Fix Diffusion Policy EMA with current Diffusers#302
ruziniuuuuu wants to merge 1 commit into
ARISE-Initiative:masterfrom
ruziniuuuuu:fix/diffusers-ema

Conversation

@ruziniuuuuu

Copy link
Copy Markdown

Summary

  • update Diffusers and Hugging Face Hub to a resolvable current cohort
  • isolate the modern EMAModel API behind a private Diffusion Policy EMA owner
  • preserve the existing checkpoint ema payload as averaged network weights
  • restore the EMA tracker from loaded averaged weights for resumed training

Validation

  • focused EMA update and checkpoint round-trip test
  • clean Python 3.12 dependency resolution with uv pip compile setup.py

This addresses the same dependency update as #263 while also fixing its checkpoint serialization and restore path.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant