aiha/
lib.rs

1#![deny(missing_docs)]
2//! # AIHA - AI Hardware Advisor
3//!
4//! ![AI Hardware Advisor](path/to/image.png)
5//!
6//! Introducing the **AI Hardware Advisor (AIHA)** – Unlock the Power of Optimal Hardware for **Your AI Project**!
7//!
8//! Looking to embark on an AI journey? Look no further! **AI Hardware Advisor (AIHA)** is the ultimate tool that empowers
9//! you to make intelligent decisions when selecting hardware for your AI endeavors.
10//!
11//! With **AIHA**, the guessing game is over. Say goodbye to uncertainty and welcome a world of precise resource allocation
12//! for inference and training any model on the esteemed Hugging Face Hub.
13//!
14pub mod hardware;
15pub mod hub;
16pub mod models;
17
18pub use hardware::{scan_hardware, Hardware, NvidiaDevice};
19pub use hub::{build_headers, ModelFile, ModelInfo, Siblings, CUSTOM_ENCODE_SET, HUB_ENDPOINT};
20pub use models::{
21    BertModelConfig, BertParams, BloomModelConfig, BloomParams, GPT2ModelConfig, GPT2Params,
22    GPTJModelConfig, GPTJParams, GPTNeoModelConfig, GPTNeoParams, LlamaModelConfig, LlamaParams,
23    ModelConfigTrait, ModelLibraries, OPTModelConfig, OPTParams, T5ModelConfig, T5Params,
24};