entrenar/hf_pipeline/fine_tune/mod.rs
1//! Fine-tuning configuration for HuggingFace models
2//!
3//! Bridges the HF pipeline with LoRA/QLoRA adapters for efficient fine-tuning.
4//!
5//! # References
6//!
7//! [1] Hu, E., et al. (2021). "LoRA: Low-Rank Adaptation of Large Language
8//! Models." arXiv:2106.09685
9//!
10//! [2] Dettmers, T., et al. (2023). "QLoRA: Efficient Finetuning of Quantized
11//! LLMs." arXiv:2305.14314
12
13mod config;
14mod memory;
15mod method;
16
17#[cfg(test)]
18mod tests;
19
20pub use config::FineTuneConfig;
21pub use memory::{MemoryRequirement, MixedPrecision};
22pub use method::FineTuneMethod;