Text Summarization - HuggingFace

This is a supervised text summarization algorithm which supports many pre-trained models available in Hugging Face. The following sample notebook demonstrates how to use the Sagemaker Python SDK for Text Summarization for using these algorithms.

For detailed documentation please refer Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK

Available Models

Model ID

Fine Tunable?

Latest Version

Min SDK Version

Source

huggingface-summarization-bart-large-cnn-samsum

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-bert-small2bert-small-finetuned-cnn-daily-mail-summarization

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-bigbird-pegasus-large-arxiv

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-bigbird-pegasus-large-pubmed

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-distilbart-cnn-12-6

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-distilbart-cnn-6-6

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-distilbart-xsum-1-1

False

1.1.0

2.75.0

HuggingFace

huggingface-summarization-distilbart-xsum-12-3

False

1.1.0

2.75.0

HuggingFace