{ | |
"base_model_name_or_path": "bigscience/bloomz-560m", | |
"encoder_hidden_size": 1024, | |
"inference_mode": true, | |
"num_attention_heads": 16, | |
"num_layers": 24, | |
"num_transformer_submodules": 1, | |
"num_virtual_tokens": 30, | |
"peft_type": "PREFIX_TUNING", | |
"prefix_projection": false, | |
"task_type": "CAUSAL_LM", | |
"token_dim": 1024 | |
} |