Skip to content

Commit 5a3d5b6

Browse files
authored
fix diffusers pipelines wan ut (#2097)
1 parent 14bceb7 commit 5a3d5b6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mindnlp/core/nn/functional.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -619,7 +619,8 @@ def interpolate(input, size=None, scale_factor=None, mode='nearest', align_corne
619619
half_pixel_centers=True)
620620
return nearest_exact(input, output_size)
621621
if input.dim() == 5 and mode == "nearest-exact":
622-
return torch._C._nn._upsample_nearest_exact3d(input, output_size, scale_factors)
622+
warnings.warn('interpolate do not support `nearest-exact` for 5-D input, use `nearest` instead.')
623+
return upsample_nearest3d_op(input, output_size, scale_factors)
623624

624625
if input.dim() == 3 and mode == "area":
625626
assert output_size is not None

0 commit comments

Comments
 (0)