[diffusion] improve: further optimize model load (#13836)

2025-12-04 18:45:20 -08:00
parent b5d3998508
commit fa0ca97694
3 changed files with 29 additions and 11 deletions
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -94,6 +94,7 @@ diffusion = [
  "st_attn ==0.0.7",
  "vsa==0.0.4",
  "yunchang==0.6.3.post1",
+  "runai_model_streamer",
 ]

 [tool.uv.extra-build-dependencies]