Downloads · 30 days
18
2% of all-time downloads
bart-base-detox-svd
s-nlp/bart-base-detox-svd
bart-base-detox-svd is a text generation model from s-nlp. Use it when you need the model to write or continue text. It is set up for transformers.
All-time downloads
730
Public
Repo size
1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin520 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- bart
- Access
- Public
- Created
- Feb 18, 2023
- Updated
- Feb 18, 2023
- SHA
- c17bd8cb
Try a prompt
Model card
From the Hugging Face model README
Model Overview
It is a SVD-compressed model of original BART-based detoxification model s-nlp/bart-base-detox.
How to use
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
model = AutoModelForSeq2SeqLM \
.from_pretrained('s-nlp/bart-base-detox-svd', trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained('facebook/bart-base')
toxics = ['that sick fuck is going to be out in 54 years.']
tokens = tokenizer(toxics)
tokens = model.generate(**tokens, num_return_sequences=1, do_sample=False,
temperature=1.0, repetition_penalty=10.0,
max_length=128, num_beams=5)
neutrals = tokenizer.decode(tokens[0, ...], skip_special_tokens=True)
print(neutrals) # stdout: She is going to be out in 54 years.
- Task
- Text Generation
- Library
- transformers
- Type
- bart
- Languages
- en
- Created
- Feb 18, 2023
- Updated
- Feb 18, 2023