Helper: create a 1-D F32 tensor from a slice.
(data: &[f32])
| 535 | |
| 536 | /// Helper: create a 1-D F32 tensor from a slice. |
| 537 | fn f32_tensor(data: &[f32]) -> Tensor { |
| 538 | Tensor::from_slice(data, (data.len(),), &Device::Cpu).unwrap() |
| 539 | } |
| 540 | |
| 541 | // ── apply_repeat_penalty_gpu ───────────────────────────────── |
| 542 |
no outgoing calls