diff --git a/lora/lora.py b/lora/lora.py
index 8a35a6d4..3eed73fb 100644
--- a/lora/lora.py
+++ b/lora/lora.py
@@ -5,13 +5,13 @@ import json
 import math
 import time
 from pathlib import Path
-from typing import List, Optional, Tuple
+from typing import List
 
 import mlx.core as mx
 import mlx.nn as nn
 import mlx.optimizers as optim
 import numpy as np
-from mlx.utils import tree_flatten, tree_map, tree_unflatten
+from mlx.utils import tree_flatten, tree_unflatten
 from models import LoRALinear, Model, ModelArgs
 from sentencepiece import SentencePieceProcessor
 
diff --git a/lora/requirements.txt b/lora/requirements.txt
index 70a195e9..4ab8df9e 100644
--- a/lora/requirements.txt
+++ b/lora/requirements.txt
@@ -1,3 +1,4 @@
 mlx>=0.0.7
 sentencepiece
 torch
+numpy
\ No newline at end of file