diff --git a/python/mlx/nn/layers/upsample.py b/python/mlx/nn/layers/upsample.py index 0be9f87bfb..b04f6e9ba7 100644 --- a/python/mlx/nn/layers/upsample.py +++ b/python/mlx/nn/layers/upsample.py @@ -302,6 +302,13 @@ def upsample_linear( indices_fn=_linear_aa_indices, align_corners=align_corners, ) + if not antialias and not align_corners: + return _interpolate_separable( + x=x, + scale_factor=scale_factor, + indices_fn=_linear_indices, + align_corners=align_corners, + ) return _interpolate( x=x, scale_factor=scale_factor,