Transformers
PyTorch
Inference Endpoints
LinhDuong's picture
Create config.json
f10b158
{
"base_model_name_or_path": "bigscience/bloomz-7b1-mt",
"bias": "none",
"enable_lora": [
true,
false,
true
],
"fan_in_fan_out": true,
"inference_mode": true,
"lora_alpha": 16,
"lora_dropout": 0.05,
"merge_weights": false,
"modules_to_save": null,
"peft_type": "LORA",
"r": 8,
"target_modules": [
"query_key_value"
],
"task_type": "CAUSAL_LM"
}