christinacdl's picture
Training in progress, epoch 1
dda1f24
raw
history blame contribute delete
568 Bytes
{
"auto_mapping": null,
"base_model_name_or_path": "mistralai/Mistral-7B-v0.1",
"inference_mode": true,
"num_attention_heads": 32,
"num_layers": 32,
"num_transformer_submodules": 1,
"num_virtual_tokens": 38,
"peft_type": "PROMPT_TUNING",
"prompt_tuning_init": "TEXT",
"prompt_tuning_init_text": "Identify if the text contains hate speech. Output '0' for 'NON-HATE' and '1' for 'HATE'. Aim to accurately classify the text.",
"revision": null,
"task_type": "SEQ_CLS",
"token_dim": 4096,
"tokenizer_name_or_path": "mistralai/Mistral-7B-v0.1"
}