From 5524ca29cc9adc92e016df7b2c202b3ce1ddbbfe Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Mon, 3 Jul 2023 17:13:23 +0100
Subject: [PATCH] Remove the fixed length hack.

---
 candle-examples/examples/bert/main.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-examples/examples/bert/main.rs b/candle-examples/examples/bert/main.rs
index 78ca97d3..5fed800d 100644
--- a/candle-examples/examples/bert/main.rs
+++ b/candle-examples/examples/bert/main.rs
@@ -607,7 +607,7 @@ fn main() -> Result<()> {
         .map_err(E::msg)?
         .get_ids()
         .to_vec();
-    let token_ids = Tensor::new(&tokens[..7], &device)?;
+    let token_ids = Tensor::new(&tokens[..], &device)?;
     println!("{token_ids}");
     let token_type_ids = token_ids.zeros_like()?;
     let ys = model.forward(&token_ids, &token_type_ids)?;