1 2 3 4 5 6 7 8 9
mod errors; mod temperature; mod filters; mod pick; pub use errors::SamplingError; pub use temperature::softmax_temperature; pub use filters::{top_k_mask, top_p_cutoff}; pub use pick::{argmax_sample, sample_from_cumulative};