thread 'main' panicked at 'called `Result::unwrap()` on an `Err` value: 
  SessionRun(Msg("

Non-zero status code returned while running Add node. 
Name:'Add_2' 
Status Message: /Users/runner/work/1/s/onnxruntime/core/providers/cpu/math/element_wise_ops.h:560 
void onnxruntime::BroadcastIterator::Append(ptrdiff_t, ptrdiff_t) axis == 1 || axis == largest was false. 
Attempting to broadcast an axis by a dimension other than 1. 4 by 77

   "))
', src/main.rs

tokenizer.with_padding(Some(PaddingParams {
    strategy: PaddingStrategy::BatchLongest,
    direction: PaddingDirection::Right,
    pad_to_multiple_of: Some(77), // <-- this right here
    pad_id: 0,
    pad_type_id: 0,
    pad_token: "[PAD]".to_string(),
}));

let v1: Vec<i32> = preprocessed
    .iter()
    .map(|i| 
        i.get_ids()
            .iter()
            .take(77) // <-- this right here
            .map(|b| *b as i32)
            .collect()
    )
    .concat();

let v2: Vec<i32> = preprocessed
    .iter()
    .map(|i| {
        i.get_attention_mask()
            .iter()
            .take(77) // <-- this right here
            .map(|b| *b as i32)
            .collect()
    })
    .concat();