diff --git a/mlx_lm/models/openlm.py b/mlx_lm/models/openlm.py
index 11be8eb..dc8df6d 100644
--- a/mlx_lm/models/openlm.py
+++ b/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