From 97939cc86ef9a3b2985b07f6cdf81a0d091673a3 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Thu, 13 Jun 2024 07:47:56 -0700 Subject: [PATCH] nits --- llms/mlx_lm/models/openlm.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llms/mlx_lm/models/openlm.py b/llms/mlx_lm/models/openlm.py index 11be8eb2..dc8df6d6 100644 --- a/llms/mlx_lm/models/openlm.py +++ b/llms/mlx_lm/models/openlm.py @@ -1,9 +1,8 @@ from dataclasses import dataclass -from typing import Dict, Optional, Tuple, Union +from typing import Optional, Tuple import mlx.core as mx import mlx.nn as nn -import numpy as np from .base import BaseModelArgs, create_additive_causal_mask