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

[Bugfix] Enable teacahce in QwenImageEditPlusPipeline (#379)

Browse Source 
Signed-off-by: yuanheng <jonathan.zhaoyh@gmail.com>
This commit is contained in:
Yuanheng Zhao authored and GitHub committed 2025-12-21 15:58:15 +08:00
1 parent 13584fd98e
commit 6b9944f11d
3 files changed
+33 -22
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

vllm_omni/diffusion/cache/teacache/extractors.py
Vendored
+1
View File
@@ -262,6 +262,7 @@ def extract_qwen_context(
EXTRACTOR_REGISTRY: dict[str, Callable] = {
"QwenImagePipeline": extract_qwen_context,
"QwenImageEditPipeline": extract_qwen_context,
"QwenImageEditPlusPipeline": extract_qwen_context,
# Future models:
# "FluxPipeline": extract_flux_context,
# "CogVideoXPipeline": extract_cogvideox_context,