Crate aiha

Source
Expand description

§AIHA - AI Hardware Advisor

AI Hardware Advisor

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.

Re-exports§

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

Modules§

hardware
Module for analyzing the hardware of the running system.
hub
Module for handling Hugging Face Hub interactions
models
Module for handling Hugging Face models