[diffusion] feat: support quant-videogen prq kv-cache quantization (memory-saving) for causal-dit (#32581)

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com> Co-authored-by: Mick <mickjagger19@icloud.com>
2026-08-08 12:56:54 +08:00
parent 24c84dfa68
commit f64328c7f6
16 changed files with 1174 additions and 54 deletions
@@ -123,6 +123,14 @@ diffusion = [
  "xatlas",
 ]

+diffusion-qvg = [
+  # quant-videogen 0.1.0 pins an incompatible Torch version, so it is
+  # installed separately with --no-deps; this extra provides its missing
+  # import-time dependencies.
+  "loguru>=0.7",
+  "termcolor>=2.3",
+]
+
 ray = [
  "ray[default]>=2.55.1",
 ]