Skip to main content

entrenar/hf_pipeline/loader/
mod.rs

1//! Model loading and teacher model abstraction
2//!
3//! Provides format-agnostic model loading with memory estimation.
4
5mod memory;
6mod safetensors;
7mod teacher;
8
9#[cfg(test)]
10mod tests;
11
12pub use memory::MemoryEstimate;
13pub use safetensors::SafeTensorsTeacher;
14pub use teacher::TeacherModel;