Skip to main content

entrenar/lora/adapter/
mod.rs

1//! LoRA adapter serialization and deserialization
2//!
3//! Enables saving and loading LoRA adapters independently of base models.
4//! This allows:
5//! - Training adapters and saving them separately
6//! - Loading adapters into frozen base models for inference
7//! - Sharing adapters without sharing full model weights
8//! - Switching between multiple adapters for the same base model
9//! - PEFT-compatible export for HuggingFace ecosystem interop
10//! - Merging adapters into base weights for deployment
11
12mod error;
13mod io;
14mod lora_adapter;
15pub(crate) mod merge_export;
16#[cfg(feature = "hub-publish")]
17pub(crate) mod merge_pipeline;
18mod metadata;
19pub(crate) mod peft_config;
20pub(crate) mod peft_export;
21
22pub use error::AdapterError;
23pub use io::{load_adapter, load_adapter_peft, save_adapter, save_adapter_peft, AdapterFormat};
24pub use lora_adapter::LoRAAdapter;
25pub use merge_export::{merge_and_collect, merge_qlora_and_collect, MergedModel};
26#[cfg(feature = "hub-publish")]
27pub use merge_pipeline::{
28    merge_export_publish, merge_qlora_export_publish, MergePublishError, MergePublishResult,
29};
30pub use metadata::AdapterMetadata;
31pub use peft_config::PeftAdapterConfig;
32pub use peft_export::PeftAdapterBundle;
33
34#[cfg(test)]
35mod tests;
36
37#[cfg(test)]
38mod falsify_tests;