aiha 0.0.4

🦉 AIHA helps you to find the minimal requirements for any model on the 🤗 Hub.
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

//! Module for handling Hugging Face models

// Base utilities for models
mod base;
pub use base::{ModelConfigTrait, ModelError, ModelLibraries};
// Bert model
pub mod bert;
pub use bert::{BertModelConfig, BertParams};
// BLOOM model
pub mod bloom;
pub use bloom::{BloomModelConfig, BloomParams};
// GPT2 model
pub mod gpt2;
pub use gpt2::{GPT2ModelConfig, GPT2Params};
// GPT-J model
pub mod gptj;
pub use gptj::{GPTJModelConfig, GPTJParams};
// GPT-Neo model
pub mod gptneo;
pub use gptneo::{GPTNeoModelConfig, GPTNeoParams};
// Llama model
pub mod llama;
pub use llama::{LlamaModelConfig, LlamaParams};
// OPT model
pub mod opt;
pub use opt::{OPTModelConfig, OPTParams};
// T5 model
pub mod t5;
pub use t5::{T5ModelConfig, T5Params};