Stash

src/inference.rs

@@ -1,4 +1,3 @@
-use std::collections::HashMap;
 use crate::node_service::{InferenceState, Tensor};
 use crate::Shard;
 

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)]