From 62dbd418d9fef992ac5d02727c8477e87bf37dd9 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Sat, 5 Oct 2024 15:24:16 -0700 Subject: [PATCH] fix tests --- llms/mlx_lm/models/deepseek.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llms/mlx_lm/models/deepseek.py b/llms/mlx_lm/models/deepseek.py index db5199e2..03cb3b1a 100644 --- a/llms/mlx_lm/models/deepseek.py +++ b/llms/mlx_lm/models/deepseek.py @@ -108,8 +108,8 @@ class DeepseekMLP(nn.Module): def __init__( self, config: ModelArgs, - hidden_size: int | None = None, - intermediate_size: int | None = None, + hidden_size: Optional[int] = None, + intermediate_size: Optional[int] = None, ): super().__init__() self.config = config