Expand description
AIHA - 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
- Module for analyzing the hardware of the running system.
- Module for handling Hugging Face Hub interactions
- Module for handling Hugging Face models