[Bugfix] Qwen-Image use teachche serve will crash (#3450)

Signed-off-by: rongfu.leng <lenronfu@gmail.com>
This commit is contained in:
rongfu.leng authored and GitHub committed 2026-05-09 06:41:59 +08:00
1 parent c2a624bec4
commit b4ab37da22
2 files changed
+41 -4

No files matched your search

+2 -1
View File
@@ -277,7 +277,8 @@ def extract_qwen_context(
def postprocess(h):
"""Apply Qwen-specific output postprocessing."""
h = module.norm_out(h, temb)
t = temb.chunk(2, dim=0)[0] if module.zero_cond_t else temb
h = module.norm_out(h, t)
output = module.proj_out(h)
if not return_dict:
return (output,)