entrenar 0.7.12

Training & Optimization library with autograd, LoRA, quantization, and model merging
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

//! Publish Configuration
//!
//! Defines the `publish:` section for auto-publishing trained models to HuggingFace Hub.

use serde::{Deserialize, Serialize};

fn default_true() -> bool {
    true
}

fn default_format() -> String {
    "safetensors".to_string()
}

/// Publish configuration for auto-uploading to HuggingFace Hub
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PublishConfig {
    /// HuggingFace repo ID (e.g., myuser/my-model)
    pub repo: String,

    /// Make the repository private
    #[serde(default)]
    pub private: bool,

    /// Generate and upload a model card
    #[serde(default = "default_true")]
    pub model_card: bool,

    /// Merge LoRA adapters before publishing
    #[serde(default)]
    pub merge_adapters: bool,

    /// Export format (safetensors or gguf)
    #[serde(default = "default_format")]
    pub format: String,
}