Downloads · 30 days
31
28% of all-time downloads
qualifier-model3-finetune-pretrained-transformer
oranne55/qualifier-model3-finetune-pretrained-transformer
qualifier-model3-finetune-pretrained-transformer is a text classification model from oranne55. Use it when you need a label for a piece of text. It is set up for transformers.
Classifies prompts as jailbreaks or benign. This is a fine-tune checkpoint of bert-base-uncased on the jailbreak-classification dataset.
All-time downloads
109
Public
Parameters
109M
3.1 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Nov 5, 2024
- Updated
- Nov 7, 2024
- SHA
- a5cc1c3d
Try a prompt
Model card
From the Hugging Face model README
Jailbreak Classifier
Classifies prompts as jailbreaks or benign. This is a fine-tune checkpoint of bert-base-uncased on the jailbreak-classification dataset.
Training Details
Training Data
Fine-tuned on the jailbreak-classification dataset.
Training Procedure
Training Hyperparameters
Second fine-tuning hyper-parameters(on train(0.8) and val(0.2))
- learning_rate = 5e-5
- train_batch_size = 8
- eval_batch_size = 8
- lr_scheduler_type = linear
- num_train_epochs = 5.0
Fecond fine-tuning hyper-parameters(on train and test)
- learning_rate = 1e-5
- train_batch_size = 8
- eval_batch_size = 8
- lr_scheduler_type = linear
- num_train_epochs = 3.0
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Nov 5, 2024
- Updated
- Nov 7, 2024