exillar25's picture
Update config.json
aab49f0 verified
{
"model_type": "mistral",
"base_model_name_or_path": "unsloth/mistral-7b-bnb-4bit",
"peft_type": "LORA",
"task_type": "CAUSAL_LM",
"inference_mode": true,
"lora": {
"r": 16,
"lora_alpha": 16,
"lora_dropout": 0,
"bias": "none",
"target_modules": [
"gate_proj",
"v_proj",
"o_proj",
"k_proj",
"down_proj",
"up_proj",
"q_proj"
],
"adapter_model": "adapter_model.safetensors"
},
"pipeline_tag": "text-generation"
}