GPU Migration Examples
Groups of models migrated to Gaudi HPU using GPU Migration toolkit.
Hugging Face Stable Diffusion
Optimum Habana Stable Diffusion
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
BLOOM 7B
Large Language Model for Inference
Hugging Face BERT large model (uncased) whole word masking
Optimum Habana BERT large model (uncased) whole word masking for PyTorch
Hugging Face RoBERTa
Optimum Habana RoBERTa for PyTorch
Hugging Face BERT
Optimum Habana BERT for PyTorch
Hugging Face ALBERT Large
Optimum Habana ALBERT Large for PyTorch
Hugging Face Swin Transformer
Optimum Habana Swin Transformer for PyTorch
Hugging Face GPT2
Optimum Habana GPT2 for PyTorch
Hugging Face ALBERT XXLarge
Optimum Habana ALBERT XXLarge for PyTorch
Hugging Face RoBERTa large
Optimum Habana RoBERTa large for PyTorch
Hugging Face DistilBERT
Optimum Habana DistilBERT for PyTorch
Hugging Face T5
Optimum Habana T5 for PyTorch
Hugging Face Vision Transformer (ViT)
Optimum Habana Vision Transformer (ViT) for PyTorch
BERT-5B
BERT model with 5 Billion Parameters using DeepSpeed for NLP