Skip to main content

entrenar/hf_pipeline/publish/
mod.rs

1//! HuggingFace Hub Publishing
2//!
3//! Publish trained models, model cards, and evaluation results to
4//! HuggingFace Hub repositories.
5//!
6//! # Example
7//!
8//! ```ignore
9//! use entrenar::hf_pipeline::publish::{HfPublisher, PublishConfig, ModelCard};
10//!
11//! let config = PublishConfig {
12//!     repo_id: "username/my-model".to_string(),
13//!     ..Default::default()
14//! };
15//! let publisher = HfPublisher::new(config)?;
16//! let result = publisher.publish(&files, Some(&card))?;
17//! println!("Published: {}", result.repo_url);
18//! ```
19
20pub mod config;
21pub mod model_card;
22pub mod publisher;
23pub mod result;
24pub mod submission;
25
26#[cfg(test)]
27mod tests;
28
29#[cfg(test)]
30mod falsification_tests;
31
32pub use config::{PublishConfig, RepoType};
33pub use model_card::ModelCard;
34pub use publisher::HfPublisher;
35pub use result::{PublishError, PublishResult};
36pub use submission::{format_submission_jsonl, format_submissions_jsonl};