BERT Fine tuning
BERT Fine Tuning Training and Inference
BERT Pretraining
BERT Pretraining for PyTorch
DistilBERT
DistilBERT model from Huggingface repository
ELECTRA Fine Tuning
ELECTRA Large Discriminator model for Fine Tuning
Hugging Face ALBERT Large
Habana Optimum ALBERT Large for PyTorch
Hugging Face ALBERT XXLarge
Habana Optimum ALBERT XXLarge for PyTorch
Hugging Face BERT
Habana Optimum BERT for PyTorch
Hugging Face BERT large model (uncased) whole word masking
Habana Optimum BERT large model (uncased) whole word masking for PyTorch
Hugging Face DistilBERT
Habana Optimum DistilBERT for PyTorch
Hugging Face GPT2
Habana Optimum GPT2 for PyTorch
Hugging Face RoBERTa
Habana Optimum RoBERTa for PyTorch
Hugging Face RoBERTa large
Habana Optimum RoBERTa large for PyTorch
Hugging Face Swin Transformer
Habana Optimum Swin Transformer for PyTorch
Hugging Face T5
Habana OptimumT5 for PyTorch
Hugging Face Vision Transformer (ViT)
Habana Optimum Vision Transformer (ViT) for PyTorch
PyTorch
These are the available PyTorch Container images in the Habana Vault.
These images contain SynapseAI SW and PyTorch framework that allow a user to run models on Gaudi.
For more details on docker usage, please refer to our installation guide on instructions to setup Container Runtime and Installation of docker images.
Stable Diffusion Models for Inference
CompVis and Stability AI based models for Stable Diffusion Inference
Stable Diffusion Training
CompVis based Stable Diffusion Model for Training
Swin Transformer
Swin Transformer – Shifted Window Model for Computer Vision
Vision Transformer
Vision Transformer – Image classification using transformers