[diffusion] fix: fix GLM-Image /v1/images/edits support (#25697)
This commit is contained in:
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user