From a6b426422e95a96bb2e63983195488d18d56ad7f Mon Sep 17 00:00:00 2001 From: Tomohiro Oga Date: Tue, 17 Dec 2024 15:30:23 +0000 Subject: [PATCH] add cubic to type hinting for upsample (#1709) --- python/mlx/nn/layers/upsample.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/upsample.py b/python/mlx/nn/layers/upsample.py index 49417a913..1f2ffd3da 100644 --- a/python/mlx/nn/layers/upsample.py +++ b/python/mlx/nn/layers/upsample.py @@ -219,7 +219,7 @@ class Upsample(Module): def __init__( self, scale_factor: Union[float, Tuple], - mode: Literal["nearest", "linear"] = "nearest", + mode: Literal["nearest", "linear", "cubic"] = "nearest", align_corners: bool = False, ): super().__init__()