anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

[Fix]: Set .transformer on SenseNova pipeline, warn if no .transformer in CacheDiT summary (#4326)

Browse Source 
Signed-off-by: Alex Brooks <albrooks@redhat.com>
Co-authored-by: Didan Deng <33117903+wtomin@users.noreply.github.com>
This commit is contained in:
Alex BrooksandDidan Deng authored and GitHub committed 2026-07-01 17:40:07 +00:00
1 parent 6b11f6b41d
commit d526dc93e8
2 files changed
+14 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/cache/cache_dit_backend.py
Vendored
+4 -1
View File
@@ -59,9 +59,12 @@ CUSTOM_DIT_ENABLERS: dict[str, Callable] = {}
# Small helper to centralize cache-dit summaries.
def cache_summary(pipeline: Any, details: bool = True) -> None:
cache_dit.summary(pipeline.transformer, details=details)
if hasattr(pipeline, "transformer"):
cache_dit.summary(pipeline.transformer, details=details)
if hasattr(pipeline, "transformer_2"):
cache_dit.summary(pipeline.transformer_2, details=details)
if not hasattr(pipeline, "transformer") and not hasattr(pipeline, "transformer_2"):
logger.warning("CacheDiT summary failed; this pipeline has no defined transformer attribute")
def default_get_pipeline_transformer(pipeline: Any) -> Any: