cargo clippy

atoma-network · Mar 27, 2024 · a159880 · a159880
1 parent 8e0fdf1
commit a159880
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/atoma-inference/src/models.rs b/atoma-inference/src/models.rs
@@ -55,9 +55,9 @@ pub enum ModelConfig {
     Llama(LlamaConfig),
     Llama2(Llama2Config),
     Mamba(MambaConfig),
-    Mixtral8x7b(MixtralConfig),
     Mistral(MistralConfig),
-    StableDiffusion(StableDiffusionConfig),
+    Mixtral8x7b(Box<MixtralConfig>),
+    StableDiffusion(Box<StableDiffusionConfig>),
 }
 
 impl From<ModelType> for ModelConfig {
@@ -74,6 +74,8 @@ pub enum ModelCache {
 
 pub trait ModelApi {
     fn load(model_specs: ModelSpecs, var_builder: VarBuilder) -> Self;
+
+    #[allow(clippy::too_many_arguments)]
     fn run(
         &self,
         input: String,