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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

//! Publishing result and error types

use std::fmt;

/// Successful publish result
#[derive(Clone, Debug)]
pub struct PublishResult {
    /// Repository URL on HuggingFace
    pub repo_url: String,
    /// Repository ID
    pub repo_id: String,
    /// Number of files uploaded
    pub files_uploaded: usize,
    /// Whether the model card was generated
    pub model_card_generated: bool,
}

impl fmt::Display for PublishResult {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(
            f,
            "Published to {} ({} files{})",
            self.repo_url,
            self.files_uploaded,
            if self.model_card_generated { " + model card" } else { "" }
        )
    }
}

/// Errors during publishing
#[derive(Debug, thiserror::Error)]
pub enum PublishError {
    /// Repository creation failed
    #[error("Failed to create repository '{repo_id}': {message}")]
    RepoCreationFailed { repo_id: String, message: String },

    /// File upload failed
    #[error("Failed to upload '{path}': {message}")]
    UploadFailed { path: String, message: String },

    /// Authentication required
    #[error("Authentication required: set HF_TOKEN or use with_token()")]
    AuthRequired,

    /// Invalid repository ID
    #[error("Invalid repository ID '{repo_id}': must be 'owner/name'")]
    InvalidRepoId { repo_id: String },

    /// HTTP error
    #[error("HTTP error: {message}")]
    Http { message: String },

    /// IO error
    #[error("IO error: {0}")]
    Io(#[from] std::io::Error),

    /// Serialization error
    #[error("Serialization error: {0}")]
    Serialization(String),
}