entrenar/hf_pipeline/export/mod.rs
1//! Model Export Formats
2//!
3//! Supports exporting trained/distilled models to various formats:
4//! - SafeTensors: Safe, fast tensor serialization
5//! - APR: Aprender format for entrenar ecosystem
6//! - GGUF: Quantized format for llama.cpp compatibility
7//!
8//! # Example
9//!
10//! ```ignore
11//! use entrenar::hf_pipeline::export::{Exporter, ExportFormat};
12//!
13//! let exporter = Exporter::new();
14//! exporter.export(&model_weights, ExportFormat::SafeTensors, "output/model.safetensors")?;
15//! ```
16
17mod exporter;
18mod format;
19pub(crate) mod gguf_verify;
20pub(crate) mod gguf_writer;
21pub(crate) mod pipeline;
22#[cfg(feature = "hub-publish")]
23pub(crate) mod publish_pipeline;
24mod result;
25mod types;
26mod weights;
27
28#[cfg(test)]
29mod tests;
30
31// Public re-exports
32pub use exporter::Exporter;
33pub use format::ExportFormat;
34pub use gguf_verify::{verify_gguf, GgufSummary, GgufTensorInfo};
35pub use gguf_writer::GgufQuantization;
36pub use pipeline::{quantize_and_export, QuantExportResult};
37#[cfg(feature = "hub-publish")]
38pub use publish_pipeline::{quantize_export_publish, QuantPublishError, QuantPublishResult};
39pub use result::ExportResult;
40#[allow(unused_imports)]
41pub use weights::TrainingMetadata;
42pub use weights::{ModelMetadata, ModelWeights};