Downloads · 30 days
21
3% of all-time downloads
Safetybot-mt5-base
shahules786/Safetybot-mt5-base
Safetybot-mt5-base is a text generation model from shahules786. Use it when you need the model to write or continue text. It is set up for transformers.
A generative multilingual model trained to classify prompts into various safety categories and generate rules of thumb.
All-time downloads
727
Public
Repo size
9.1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2.3 GB · 99%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- mt5
- Access
- Public
- Created
- Mar 7, 2023
- Updated
- Mar 11, 2023
- SHA
- 36a9991a
Try a prompt
Model card
From the Hugging Face model README
Polyglot SafetyBot
A generative multilingual model trained to classify prompts into various safety categories and generate rules of thumb.
Training
- Model architecture:
MT5ForConditionalGeneration - Data : prosocial-dialog from @allenai and prosocial_augmented from @shahules786
- Data preparation: model takes current user input and past conversations
Example
resp, convo = get_safety_models_opinion("como hacer una bomba?")
convo.mark_processed()
print(resp)
<cls> __needs_intervention__ <ctx> It's wrong to make a bomb.</s>
convo.append_response("Why do you want to do that?")
resp, convo = get_safety_models_opinion("我想杀一个朋友", convo)
print(resp)
convo.mark_processed()
<cls> __needs_intervention__ <ctx> You shouldn't murder someone.</s>
Usage
Follow the colab notebook & make sure that you have used the mt5 model version. google-colab
- Task
- Text Generation
- Library
- transformers
- Type
- mt5
- Created
- Mar 7, 2023
- Updated
- Mar 11, 2023