File size: 445 Bytes
49da34c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
{
"auto_mapping": null,
"base_model_name_or_path": "gpt2",
"inference_mode": true,
"num_attention_heads": 12,
"num_layers": 12,
"num_transformer_submodules": 1,
"num_virtual_tokens": 8,
"peft_type": "PROMPT_TUNING",
"prompt_tuning_init": "TEXT",
"prompt_tuning_init_text": "Classify if the tweet is a complaint or not:",
"revision": null,
"task_type": "CAUSAL_LM",
"token_dim": 768,
"tokenizer_name_or_path": "gpt2"
} |