entrenar 0.7.9

Training & Optimization library with autograd, LoRA, quantization, and model merging
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

//! Teacher model configuration

use serde::{Deserialize, Serialize};

/// Teacher model configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TeacherConfig {
    /// Model ID on HuggingFace
    pub model_id: String,
    /// Revision/branch (default: "main")
    #[serde(default = "default_revision")]
    pub revision: String,
    /// Use 8-bit quantization for teacher
    #[serde(default)]
    pub load_in_8bit: bool,
}

pub(crate) fn default_revision() -> String {
    "main".to_string()
}