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

#![deny(missing_docs)]
//! # AIHA - AI Hardware Advisor
//!
//! ![AI Hardware Advisor](path/to/image.png)
//!
//! Introducing the **AI Hardware Advisor (AIHA)** – Unlock the Power of Optimal Hardware for **Your AI Project**!
//!
//! Looking to embark on an AI journey? Look no further! **AI Hardware Advisor (AIHA)** is the ultimate tool that empowers
//! you to make intelligent decisions when selecting hardware for your AI endeavors.
//!
//! With **AIHA**, the guessing game is over. Say goodbye to uncertainty and welcome a world of precise resource allocation
//! for inference and training any model on the esteemed Hugging Face Hub.
//!
pub mod hardware;
pub mod hub;
pub mod models;

pub use hardware::{scan_hardware, Hardware, NvidiaDevice};
pub use hub::{build_headers, ModelFile, ModelInfo, Siblings, CUSTOM_ENCODE_SET, HUB_ENDPOINT};
pub use models::{
    BertModelConfig, BertParams, BloomModelConfig, BloomParams, GPT2ModelConfig, GPT2Params,
    GPTJModelConfig, GPTJParams, GPTNeoModelConfig, GPTNeoParams, LlamaModelConfig, LlamaParams,
    ModelConfigTrait, ModelLibraries, OPTModelConfig, OPTParams, T5ModelConfig, T5Params,
};