sidnb13's picture
Upload model
a9b8c98
raw
history blame contribute delete
456 Bytes
{
"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": "Determine what type of institution the research paper is from:",
"task_type": "CAUSAL_LM",
"token_dim": 1024,
"tokenizer_name_or_path": "bigscience/bloomz-560m"
}