[Bugfix] Qwen-Image use teachche serve will crash (#3450)
Signed-off-by: rongfu.leng <lenronfu@gmail.com>
This commit is contained in:
1 parent
c2a624bec4
commit
b4ab37da22
2 files changed
+41
-4
No files matched your search
+2
-1
@@ -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,)
|
||||
|
||||
Reference in new issue
Block a user