joshuacoles/exo-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Stash

Browse Source
This commit is contained in:
Joshua Coles 2025-02-13 09:34:12 +00:00
parent e551efc0a4
commit 4989c7b2f3
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/inference.rs
View File

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

1
src/llama_module.rs
View File

@ -6,7 +6,6 @@ use mlx_rs::nn::{Embedding, RmsNorm, RmsNormBuilder};
use mlx_rs::Array; use mlx_rs::Array;
use serde::Deserialize; use serde::Deserialize;
use std::collections::HashMap; use std::collections::HashMap;
use std::env::args;
use mlx_rs::ops::zeros; use mlx_rs::ops::zeros;
#[derive(Debug, Deserialize, ModuleParameters)] #[derive(Debug, Deserialize, ModuleParameters)]