clippy and fmt

src/common/mod.rs

@@ -134,6 +134,7 @@ impl MambaWrapper {
 
                 // if the token has some valid representation, print it
                 if let Some(t) = self.tokenizer.next_token(next_token)? {
+                    #[allow(unused_imports)]
                     use std::io::Write;
                     print!("{t}");
                     std::io::stdout().flush()?;
@@ -177,6 +178,7 @@ impl MambaWrapper {
 
             // if the token has some valid representation, print it
             if let Some(t) = self.tokenizer.next_token(next_token)? {
+                #[allow(unused_imports)]
                 use std::io::Write;
                 print!("{t}");
                 std::io::stdout().flush()?;

src/common/token_output_stream.rs

@@ -85,4 +85,4 @@ impl TokenOutputStream {
         self.prev_index = 0;
         self.current_index = 0;
     }
-}
+}

src/native/main.rs

@@ -2,18 +2,14 @@
 
 // #![allow(clippy::erasing_op)]
 
-use hf_hub::types::FilePath;
-use mamba_minimal_dfdx_example::{hf, mamba, LogitsProcessorWrapper, MambaWrapper};
-
-// use candle_examples::token_output_stream::TokenOutputStream;
-use candle_transformers::generation::LogitsProcessor;
 use dfdx::prelude::*;
+use hf_hub::types::FilePath;
 use hf_hub::{
     api::sync::Api,
     types::{RepoId, RevisionPath},
     Repo, RepoType,
 };
-use tokenizers::Tokenizer;
+use mamba_minimal_dfdx_example::{hf, mamba, LogitsProcessorWrapper, MambaWrapper};
 
 fn main() -> anyhow::Result<()> {
     let start = std::time::Instant::now();

src/wasm/non_ui.rs

@@ -1,17 +1,11 @@
-use core::slice::SlicePattern;
-
-use crate::{hf, mamba, token_output_stream};
+use crate::{hf, mamba};
 use crate::{LogitsProcessorWrapper, MambaWrapper};
-use candle_transformers::generation::LogitsProcessor;
 use dfdx::prelude::*;
 use hf_hub::{
     api::wasm::Api,
     types::{FilePath, RepoId, RevisionPath},
     Repo, RepoType,
 };
-use token_output_stream::TokenOutputStream;
-use tokenizers::Tokenizer;
-use wasm_bindgen::prelude::wasm_bindgen;
 
 pub async fn run() -> anyhow::Result<()> {
     let api = Api::new().await?;

src/wasm/yew_ui/model.rs

@@ -1,17 +1,11 @@
+use crate::{hf, mamba, LogitsProcessorWrapper, MambaWrapper};
 use dfdx::tensor::Cpu;
 use hf_hub::{
     api::wasm::{Api, ApiRepo, Metadata, UrlTemplate},
-    types::{
-        Endpoint, FilePath, FileUrl, RepoId, RevisionPath, TmpFileBlobKey, TmpFileBlobKeyList,
-    },
+    types::{Endpoint, FilePath, FileUrl, RepoId, RevisionPath, TmpFileBlobKeyList},
     Repo, RepoType,
 };
 use tokenizers::Tokenizer;
-use yew::prelude::*;
-
-use crate::{
-    hf, mamba, token_output_stream::TokenOutputStream, LogitsProcessorWrapper, MambaWrapper,
-};
 
 pub struct Model {
     // general data

src/wasm/yew_ui/update.rs

@@ -1,7 +1,6 @@
 use super::model::ModelSelection;
 pub use super::model::{self, Connection, Model};
 use super::Msg;
-use dfdx::{nn_traits::ResetParams, tensor::Cpu};
 use hf_hub::{api::wasm::Api, types::TmpFileBlobKey};
 use yew::prelude::*;
 
@@ -102,10 +101,10 @@ impl model::Model {
                 model_data.cache.is_done = true;
                 true
             }
-            Msg::StartModelDataUpload(selection) => {
+            Msg::StartModelDataUpload(_selection) => {
                 todo!()
             }
-            Msg::FinishModelDataUpload(selection) => {
+            Msg::FinishModelDataUpload(_selection) => {
                 todo!()
             }
             Msg::StartModelDataLoad(selection) => {