mlmf 0.2.0

Machine Learning Model Files - Loading, saving, and dynamic mapping for ML models
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
55
56
57
58
59
60
61
62
63

//! Format-specific model loaders
//!
//! This module contains specialized loaders for different model file formats.
//! Each format has its own specific requirements and optimizations.

pub mod safetensors;

#[cfg(feature = "gguf")]
pub mod gguf;

#[cfg(feature = "gguf")]
pub mod gguf_export;

#[cfg(feature = "onnx")]
pub mod onnx_export;

#[cfg(feature = "onnx")]
pub mod onnx_import;

#[cfg(feature = "awq")]
pub mod awq;

#[cfg(feature = "pytorch")]
pub mod pytorch_loader;

// Export modules
pub mod safetensors_export;

#[cfg(feature = "pytorch")]
pub mod pytorch_export;

#[cfg(feature = "awq")]
pub mod awq_export;

// Re-export commonly used types
pub use safetensors::*;

#[cfg(feature = "gguf")]
pub use gguf::*;

#[cfg(feature = "gguf")]
pub use gguf_export::*;

#[cfg(feature = "onnx")]
pub use onnx_export::*;

#[cfg(feature = "onnx")]
pub use onnx_import::*;

#[cfg(feature = "awq")]
pub use awq::*;

#[cfg(feature = "pytorch")]
pub use pytorch_loader::*;

// Re-export export functions
pub use safetensors_export::*;

#[cfg(feature = "pytorch")]
pub use pytorch_export::*;

#[cfg(feature = "awq")]
pub use awq_export::*;