distilbart cnn 6 6
Overview
Highlights
- Optimized for fast, low-latency text summarization tasks
- Reduced memory footprint for efficient edge deployment
- Full compatibility with Hugging Face Transformers library
- Maintains high accuracy through advanced knowledge distillation
- Apache-2.0 license ensures flexible commercial integration
Usage
# Install Hugging Face transformers
pip install transformers torch
# Load model with transformers
from transformers import AutoModel, AutoTokenizer
model = AutoModel.from_pretrained("Xenova/distilbart-cnn-6-6")
tokenizer = AutoTokenizer.from_pretrained("Xenova/distilbart-cnn-6-6")
Hugging Face Download
We recommend downloading the model via the Hugging Face CLI or Hub SDK.
Guidance:Before downloading, install huggingface_hub with:
pip install -U huggingface_hub
CLI Download
Download the full repository
huggingface-cli download Xenova/distilbart-cnn-6-6
Download a single file to a local folder (e.g. config.json into ./dir)
huggingface-cli download Xenova/distilbart-cnn-6-6 config.json --local-dir ./dir
See the official docs for more CLI options
SDK Download
# 模型下载
from huggingface_hub import snapshot_download
model_dir = snapshot_download('Xenova/distilbart-cnn-6-6')
Git Download
Make sure git-lfs is installed first
git lfs install
git clone https://huggingface.co/Xenova/distilbart-cnn-6-6
To skip LFS large-file downloads, use:
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/Xenova/distilbart-cnn-6-6
Model files are hosted on the Hugging Face Hub — download directly via HF CLI / SDK / Git, not through this site.
PyTorch / Transformers Usage
Install Transformers
pip install -U transformers torch
Load the model and run inference
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained('Xenova/distilbart-cnn-6-6')
tokenizer = AutoTokenizer.from_pretrained('Xenova/distilbart-cnn-6-6')
Model Download
We recommend downloading the model via the ModelScope CLI or SDK.
Guidance:Before downloading, install ModelScope with:
pip install modelscope
CLI Download
Download the full repository
modelscope download --model Xenova/distilbart-cnn-6-6
Download a single file to a local folder (e.g. README.md into ./dir)
modelscope download --model Xenova/distilbart-cnn-6-6 README.md --local_dir ./dir
See the docs for more CLI options
SDK Download
# 模型下载
from modelscope import snapshot_download
model_dir = snapshot_download('Xenova/distilbart-cnn-6-6')
Git Download
Make sure git-lfs is installed first
git lfs install
git clone https://www.modelscope.cn/Xenova/distilbart-cnn-6-6.git
To skip LFS large-file downloads, use:
GIT_LFS_SKIP_SMUDGE=1 git clone https://www.modelscope.cn/Xenova/distilbart-cnn-6-6.git
ModelScope 模型页直接下载模型文件;无需将模型文件放在本站服务器。
Notebook Quickstart
Install the ModelScope library
pip install "modelscope[audio,cv,nlp,multi-modal,science]" -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html
Load the model and run inference
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
p = pipeline('text-generation', 'Xenova/distilbart-cnn-6-6')
Full Documentation
---
base_model: sshleifer/distilbart-cnn-6-6
library_name: transformers.js
license: apache-2.0
pipeline_tag: summarization
---
https://huggingface.co/sshleifer/distilbart-cnn-6-6 with ONNX weights to be compatible with Transformers.js.
Usage (Transformers.js)
If you haven't already, you can install the Transformers.js JavaScript library from NPM using:
npm i @huggingface/transformersExample: Summarization.
import { pipeline } from '@huggingface/transformers';
const generator = await pipeline('summarization', 'Xenova/distilbart-cnn-6-6');
const text = 'The tower is 324 metres (1,063 ft) tall, about the same height as an 81-storey building, ' +
'and the tallest structure in Paris. Its base is square, measuring 125 metres (410 ft) on each side. ' +
'During its construction, the Eiffel Tower surpassed the Washington Monument to become the tallest ' +
'man-made structure in the world, a title it held for 41 years until the Chrysler Building in New ' +
'York City was finished in 1930. It was the first structure to reach a height of 300 metres. Due to ' +
'the addition of a broadcasting aerial at the top of the tower in 1957, it is now taller than the ' +
'Chrysler Building by 5.2 metres (17 ft). Excluding transmitters, the Eiffel Tower is the second ' +
'tallest free-standing structure in France after the Millau Viaduct.';
const output = await generator(text, {
max_new_tokens: 100,
});
Note: Having a separate repo for ONNX weights is intended to be a temporary solution until WebML gains more traction. If you would like to make your models web-ready, we recommend converting to ONNX using 🤗 Optimum and structuring your repo like this one (with ONNX weights located in a subfolder named onnx).