[diffusion] feat: support SparseVideoGen2 attention backend (#17507)

Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Xinwei Qiang
2026-02-13 16:20:46 +08:00
committed by GitHub
parent d97eb111a3
commit 356e338607
9 changed files with 732 additions and 23 deletions

View File

@@ -148,7 +148,10 @@ def _log_process_aware(
if should_log:
# stacklevel=3 to show the original caller's location,
# as this function is called by the patched methods.
logger_self.log(level, msg, *args, stacklevel=3, **kwargs)
if "stacklevel" in kwargs:
logger_self.log(level, msg, *args, **kwargs)
else:
logger_self.log(level, msg, *args, stacklevel=3, **kwargs)
class _SGLDiffusionLogger(Logger):