Skip to main content

entrenar/hf_pipeline/config/
teacher.rs

1//! Teacher model configuration
2
3use serde::{Deserialize, Serialize};
4
5/// Teacher model configuration
6#[derive(Debug, Clone, Serialize, Deserialize)]
7pub struct TeacherConfig {
8    /// Model ID on HuggingFace
9    pub model_id: String,
10    /// Revision/branch (default: "main")
11    #[serde(default = "default_revision")]
12    pub revision: String,
13    /// Use 8-bit quantization for teacher
14    #[serde(default)]
15    pub load_in_8bit: bool,
16}
17
18pub(crate) fn default_revision() -> String {
19    "main".to_string()
20}