Update Sequential to implement forward_t instead of forward.

huggingface · Dec 31, 2023 · 66531aa · 66531aa
1 parent cf51ea7
commit 66531aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/candle-nn/src/sequential.rs b/candle-nn/src/sequential.rs
@@ -24,10 +24,10 @@ impl Sequential {
 }
 
 impl Module for Sequential {
-    fn forward(&self, xs: &Tensor) -> Result<Tensor> {
+    fn forward_t(&self, xs: &Tensor, train: bool) -> Result<Tensor> {
         let mut xs = xs.clone();
         for layer in self.layers.iter() {
-            xs = layer.forward(&xs)?
+            xs = layer.forward_t(&xs, train)?;
         }
         Ok(xs)
     }