Downloads · 30 days
8
12% of all-time downloads
lidiya_-_bart-large-xsum-samsum-4bits
RichardErkhov/lidiya_-_bart-large-xsum-samsum-4bits
lidiya_-_bart-large-xsum-samsum-4bits is a text generation model from RichardErkhov. Use it when you need the model to write or continue text. It is set up for transformers.
All-time downloads
69
Public
Parameters
257M
219 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors219 MB · 98%
Parameter types
How the weights are stored.
U8201M · 78%
Try a prompt
Model card
From the Hugging Face model README
Quantization made by Richard Erkhov.
bart-large-xsum-samsum - bnb 4bits
- Model creator: https://huggingface.co/lidiya/
- Original model: https://huggingface.co/lidiya/bart-large-xsum-samsum/
Original model description:
language: en tags:
- bart
- seq2seq
- summarization license: apache-2.0 datasets:
- samsum widget:
- text: | Hannah: Hey, do you have Betty's number? Amanda: Lemme check Amanda: Sorry, can't find it. Amanda: Ask Larry Amanda: He called her last time we were at the park together Hannah: I don't know him well Amanda: Don't be shy, he's very nice Hannah: If you say so.. Hannah: I'd rather you texted him Amanda: Just text him 🙂 Hannah: Urgh.. Alright Hannah: Bye Amanda: Bye bye model-index:
- name: bart-large-xsum-samsum
results:
- task:
name: Abstractive Text Summarization
type: abstractive-text-summarization
dataset:
name: "SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarization"
type: samsum
metrics:
- name: Validation ROUGE-1 type: rouge-1 value: 54.3921
- name: Validation ROUGE-2 type: rouge-2 value: 29.8078
- name: Validation ROUGE-L type: rouge-l value: 45.1543
- name: Test ROUGE-1 type: rouge-1 value: 53.3059
- name: Test ROUGE-2 type: rouge-2 value: 28.355
- name: Test ROUGE-L type: rouge-l value: 44.0953
- task:
name: Abstractive Text Summarization
type: abstractive-text-summarization
dataset:
name: "SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarization"
type: samsum
metrics:
bart-large-xsum-samsum
This model was obtained by fine-tuning facebook/bart-large-xsum on Samsum dataset.
Usage
from transformers import pipeline
summarizer = pipeline("summarization", model="lidiya/bart-large-xsum-samsum")
conversation = '''Hannah: Hey, do you have Betty's number?
Amanda: Lemme check
Amanda: Sorry, can't find it.
Amanda: Ask Larry
Amanda: He called her last time we were at the park together
Hannah: I don't know him well
Amanda: Don't be shy, he's very nice
Hannah: If you say so..
Hannah: I'd rather you texted him
Amanda: Just text him 🙂
Hannah: Urgh.. Alright
Hannah: Bye
Amanda: Bye bye
'''
summarizer(conversation)
Training procedure
- Colab notebook: https://colab.research.google.com/drive/1dul0Sg-TTMy9xZCJzmDRajXbyzDwtYx6?usp=sharing
Results
| key | value |
|---|---|
| eval_rouge1 | 54.3921 |
| eval_rouge2 | 29.8078 |
| eval_rougeL | 45.1543 |
| eval_rougeLsum | 49.942 |
| test_rouge1 | 53.3059 |
| test_rouge2 | 28.355 |
| test_rougeL | 44.0953 |
| test_rougeLsum | 48.9246 |
- Task
- Text Generation
- Library
- transformers
- Type
- bart
- Created
- May 9, 2024
- Updated
- May 9, 2024