This repository has been archived by the owner on Jun 6, 2024. It is now read-only.
Initial release of infernet-ml, a library to build Web3 machine learning workflows.
Features
- Workflow classes for running inference:
HFInferenceClientWorkflow
: Wrapper around Huggingface's Inference Client libraryONNXInferenceWorkflow
: ONNX model suportTorchInferenceWorkflow
: PyTorch model supportTTSInferenceWorkflow
: Base class for text-to-speech model supportTGIClientInferenceWorkflow
: Huggingface's Text Generation Interface server inference supportBarkHFInferenceWorkflow
: Suno's Bark model support
- Utility for model retrieval from different sources, beginning with:
- Hugginface Hub
- Arweave