diff --git a/src/inference.rs b/src/inference.rs
index 136d443..cea4d14 100644
--- a/src/inference.rs
+++ b/src/inference.rs
@@ -1,4 +1,3 @@
-use std::collections::HashMap;
 use crate::node_service::{InferenceState, Tensor};
 use crate::Shard;
 
diff --git a/src/llama_module.rs b/src/llama_module.rs
index 591484d..b432880 100644
--- a/src/llama_module.rs
+++ b/src/llama_module.rs
@@ -6,7 +6,6 @@ use mlx_rs::nn::{Embedding, RmsNorm, RmsNormBuilder};
 use mlx_rs::Array;
 use serde::Deserialize;
 use std::collections::HashMap;
-use std::env::args;
 use mlx_rs::ops::zeros;
 
 #[derive(Debug, Deserialize, ModuleParameters)]