distill.rs
1use std::mem;
2
3use crate::example::Example;
4
5pub async fn run_distill(example: &mut Example) {
6 let [prediction]: [_; 1] = mem::take(&mut example.predictions)
7 .try_into()
8 .expect("Run predict first with a single repetition");
9
10 example.expected_patch = prediction.actual_patch;
11 example.prompt = None;
12 example.predictions = Vec::new();
13 example.score = Vec::new();
14}