jackhhao/jailbreak-classifier

Model

24

stars

6

commits

10

repos using this model

2

linked in READMEs

Apr 5, 2024

updated

bert
endpoints_compatible
jailbreak
moderation
prompt-injection
pytorch
security
text-classification
text-embeddings-inference
transformers
Browse cluster: Transformer Text Classification & Embeddings

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

Fine-tuning hyper-parameters:

  • learning_rate = 5e-5
  • train_batch_size = 8
  • eval_batch_size = 8
  • lr_scheduler_type = linear
  • num_train_epochs = 5.0

Contributors

jackhhao

6 commits

jackhhao/jailbreak-classifier

Model

24

stars

6

commits

10

repos using this model

2

linked in READMEs

Apr 5, 2024

updated

bert
endpoints_compatible
jailbreak
moderation
prompt-injection
pytorch
security
text-classification
text-embeddings-inference
transformers
Browse cluster: Transformer Text Classification & Embeddings

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

Fine-tuning hyper-parameters:

  • learning_rate = 5e-5
  • train_batch_size = 8
  • eval_batch_size = 8
  • lr_scheduler_type = linear
  • num_train_epochs = 5.0

Contributors

jackhhao

6 commits