[XPU] Make checkpoint_engine worker device-agnostic (#32382)

2026-09-11 09:56:39 +08:00
parent ad7f57c9ea
commit 67d3a2ea57
5 changed files with 171 additions and 13 deletions
@@ -73,6 +73,7 @@ dependencies = [
 ]

 [project.optional-dependencies]
+checkpoint-engine = ["checkpoint-engine @ git+https://github.com/MoonshotAI/checkpoint-engine.git@f772d1858204fa82bdb0436fdfec137f5de2972c"]
 diffusion = [
  "addict==2.4.0",
  "av==16.1.0",