mold-ai-core 0.12.0

Shared types, API protocol, and HTTP client for mold
Documentation 
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

pub mod build_info;
pub mod catalog;
pub mod chain;
pub mod chain_toml;
pub mod client;
pub mod config;
pub mod control;
pub mod download;
pub mod error;
pub mod expand;
pub mod expand_prompts;
pub mod install_error;
pub mod lambda;
pub mod manifest;
pub mod media_paths;
pub mod runpod;
pub mod types;
pub mod validation;

#[cfg(test)]
mod config_test;
#[cfg(test)]
mod test_support;

pub use catalog::build_model_catalog;
pub use chain::{
    ChainFailure, ChainProgressEvent, ChainRequest, ChainResponse, ChainScript, ChainScriptChain,
    ChainStage, LoraSpec, NamedRef, SseChainCompleteEvent, TransitionMode, VramEstimate,
    MAX_CHAIN_STAGES,
};
pub use client::MoldClient;
pub use config::{
    parse_device_ref_str, Config, DefaultModelResolution, DefaultModelSource, LoggingConfig,
    ModelConfig, ModelPaths,
};
pub use control::{
    classify_generate_error, classify_server_error, GenerateServerAction, ServerAvailability,
};
pub use error::{MoldError, Result as MoldResult};
pub use install_error::InstallError;
pub use media_paths::{configured_media_roots, parse_media_roots_env, resolve_server_media_path};
pub use types::GenerateRequest;
pub use types::Scheduler;
pub use types::*;
pub use validation::{
    clamp_to_megapixel_limit, dimension_warning, family_supports_lora, fit_to_model_dimensions,
    fit_to_target_area, recommended_dimensions, validate_generate_request,
    validate_generate_request_with_family, validate_upscale_request, LORA_CAPABLE_FAMILIES,
};

pub use expand::{
    ApiExpander, ExpandConfig, ExpandResult, ExpandSettings, FamilyOverride, PromptExpander,
};
pub use expand_prompts::{build_batch_messages, build_single_messages, format_chatml};