huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove the fixed length hack.

Browse Source
This commit is contained in:
laurent
2023-07-03 17:13:23 +01:00
parent 1ea6690557
commit 5524ca29cc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/bert/main.rs
View File

@ -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)?;