[diffusion] fix: fix GLM-Image /v1/images/edits support (#25697)

This commit is contained in:
Chi McIsaac
2026-05-20 17:11:51 +08:00
committed by GitHub
parent 614672fea5
commit 47979fb252
4 changed files with 101 additions and 28 deletions
@@ -343,6 +343,8 @@ def fuse_scale_shift_kernel(
B, L, C = x.shape
output = torch.empty_like(x)
if x.numel() == 0:
return output
if scale.dim() == 4:
# scale/shift: [B, F, 1, C]