PyPI page
Home page
Author:
HuggingFace Inc. Special Ops Team
License:
Apache
Summary:
Optimum Habana is the interface between the Hugging Face Transformers and Diffusers libraries and Habana's Gaudi processor (HPU). It provides a set of tools enabling easy model loading, training and inference on single- and multi-HPU settings for different downstream tasks.
Latest version:
1.14.1
Required dependencies:
accelerate
|
diffusers
|
huggingface-hub
|
optimum
|
sentence-transformers
|
torch
|
transformers
Optional dependencies:
datasets
|
gitpython
|
hf-doc-builder
|
optuna
|
parameterized
|
peft
|
psutil
|
pytest
|
ruff
|
safetensors
|
scipy
|
sentencepiece
|
timm
|
torchsde
Downloads last day:
61
Downloads last week:
2,338
Downloads last month:
13,918