Fix set_dtype

huggingface · Oct 3, 2024 · 20a57c4 · 20a57c4
1 parent 156ebd1
commit 20a57c4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/candle-nn/src/var_builder.rs b/candle-nn/src/var_builder.rs
@@ -236,6 +236,7 @@ impl<'a, B: Backend> VarBuilderArgs<'a, B> {
                 dtype,
                 device: self.data.device.clone(),
             }),
+            dtype,
             ..self
         }
     }