Format

Author: EricLBuehler

candle-nn/src/cpu_flash_attention.rs

@@ -68,7 +68,7 @@ fn vec_dot<T: WithDType + Sum + Copy + std::ops::Mul<Output = T>>(a: &[T], b: &[
 /// - `k`: (bs, kv_seq, v_head, hidden)
 /// - `k`: (bs, kv_seq, kv_head_seq, v_hidden)
 /// - `scale` is applied before softmax.
-/// 
+///
 /// - This supports ALiBi with `max_bias` as well as softcapping with `softcap`.
 ///
 /// **Output shape:** (bs, seq, qhead, v_hidden)