Skip to content

Commit 75e4795

Browse files
authored
Merge pull request #6 from second-state/fix-where-self-arg-order
Fix where_self argument order in attention mask construction
2 parents eed686e + 0661e62 commit 75e4795

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/audio_encoder.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ impl AudioEncoder {
236236
);
237237
// where(allow, 0, -inf)
238238
let mask = Tensor::from_tch(
239-
allow_mask.as_tch().where_self(&zero.into_tch(), &neg_inf.into_tch())
239+
zero.into_tch().where_self(&allow_mask.as_tch(), &neg_inf.into_tch())
240240
);
241241
Some(mask)
242242
}

0 commit comments

Comments
 (0)