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.21.0
Required dependencies:
accelerate
|
diffusers
|
huggingface-hub
|
optimum
|
sentence-transformers
|
sentencepiece
|
torch
|
transformers
Optional dependencies:
bitsandbytes
|
blobfile
|
datasets
|
gitpython
|
optuna
|
parameterized
|
peft
|
psutil
|
pytest
|
ruff
|
safetensors
|
scipy
|
sentencepiece
|
tiktoken
|
torchsde
Downloads last day:
239
Downloads last week:
1,814
Downloads last month:
8,263