clippy fixes

tomsanbear · Apr 3, 2024 · 4628f85 · 4628f85
1 parent 7b56d82
commit 4628f85
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/bit_transformer.rs b/src/bit_transformer.rs
@@ -65,7 +65,7 @@ impl BitTransformer {
     #[instrument]
     pub fn forward(&self, x: &Tensor) -> Result<Tensor> {
         // Run the embedding layer
-        let x_embed = self.embedding.forward(&x)?;
+        let x_embed = self.embedding.forward(x)?;
 
         // Fold each block forward
         let mut x = x_embed.clone();

diff --git a/src/training.rs b/src/training.rs
@@ -71,7 +71,7 @@ pub fn run(args: &TrainingCmd, common_args: &Args) -> Result<()> {
     let vb = VarBuilder::from_varmap(&varmap, dtype, &device);
 
     // Get the datasets
-    let dataset = { Dataset::new(args.dataset.to_string())? };
+    let dataset = { Dataset::new(&args.dataset)? };
 
     // Setup the model
     let config = Config {