File size: 438 Bytes
f937d0d
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
{
  "base_model_name_or_path": "bigscience/bloomz-560m",
  "inference_mode": true,
  "num_attention_heads": 16,
  "num_layers": 24,
  "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:",
  "task_type": "CAUSAL_LM",
  "token_dim": 1024,
  "tokenizer_name_or_path": "bigscience/bloomz-560m"
}