trustformers-models 0.1.1

Model implementations for TrustformeRS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

pub mod config;
pub mod model;
pub mod tasks;
pub mod tools;

#[cfg(test)]
mod tests;

pub use config::MistralV3Config;
pub use model::{
    MistralV3Attention, MistralV3DecoderLayer, MistralV3ForCausalLM, MistralV3MLP, MistralV3Model,
    MistralV3RmsNorm,
};
pub use tasks::{
    format_tool_call_prompt, parse_tool_call_response, MistralV3Error, MistralV3WithTools,
    ToolCallRequest, ToolDefinition as MistralV3ToolDefinition,
    ToolParameter as MistralV3ToolParameter, ToolUseTokens,
};
pub use tools::{MistralV3FunctionCaller, ToolCall, ToolDefinition, ToolParameter, ToolParseError};