Remove unused imports

Author: odysa

mlx_lm/benchmark.py

@@ -7,7 +7,7 @@
 
 from mlx_lm import batch_generate, load, stream_generate
 from mlx_lm.generate import DEFAULT_MODEL
-from mlx_lm.utils import pipeline_load, sharded_load
+from mlx_lm.utils import sharded_load
 
 
 def setup_arg_parser():

mlx_lm/chat_templates/deepseek_v32.py

@@ -2,7 +2,6 @@
 
 import copy
 import json
-import re
 from inspect import isfunction
 from typing import Any, Dict, List, Optional, Tuple, Union
 

mlx_lm/examples/tool_use.py

@@ -1,7 +1,5 @@
 # Copyright © 2025 Apple Inc.
 
-import json
-
 from mlx_lm import generate, load
 from mlx_lm.models.cache import make_prompt_cache
 

mlx_lm/generate.py

@@ -9,7 +9,6 @@
 import time
 from collections import deque
 from dataclasses import dataclass
-from functools import partial
 from typing import (
     Any,
     Callable,

mlx_lm/lora.py

@@ -3,7 +3,6 @@
 import os
 import re
 import types
-import warnings
 from pathlib import Path
 
 import mlx.core as mx

mlx_lm/models/afm7.py

@@ -4,7 +4,7 @@
 from dataclasses import dataclass
 from functools import partial
 from itertools import accumulate
-from typing import Any, Dict, Optional, Union
+from typing import Any, Optional
 
 import mlx.core as mx
 import mlx.nn as nn

mlx_lm/models/apertus.py

@@ -1,7 +1,6 @@
 # Copyright © 2023-2025 Apple Inc.
 
 from dataclasses import dataclass
-from functools import partial
 from typing import Any, Dict, Optional, Union
 
 import mlx.core as mx

mlx_lm/models/base.py

@@ -2,7 +2,7 @@
 
 import inspect
 from dataclasses import dataclass
-from typing import Any, Optional
+from typing import Optional
 
 import mlx.core as mx
 from mlx.utils import tree_map

mlx_lm/models/bitlinear_layers.py

@@ -2,7 +2,6 @@
 
 import mlx.core as mx
 import mlx.nn as nn
-from mlx.nn.layers.quantized import QuantizedLinear
 from mlx.utils import tree_flatten, tree_unflatten
 
 

mlx_lm/models/bitnet.py

@@ -1,7 +1,6 @@
 # Copyright © 2023-2024 Apple Inc.
 
 from dataclasses import dataclass
-from functools import partial
 from typing import Any, Dict, Optional, Union
 
 import mlx.core as mx