Quentin Meeus
Finetune E2E NER model for 5000 steps with LoRA
9aefa35
raw
history blame
734 Bytes
{
"alpha_pattern": {},
"auto_mapping": {
"base_model_class": "WhisperForConditionalGeneration",
"parent_library": "transformers.models.whisper.modeling_whisper"
},
"base_model_name_or_path": "openai/whisper-small",
"bias": "none",
"fan_in_fan_out": false,
"inference_mode": true,
"init_lora_weights": true,
"layers_pattern": null,
"layers_to_transform": null,
"loftq_config": {},
"lora_alpha": 32,
"lora_dropout": 0.1,
"megatron_config": null,
"megatron_core": "megatron.core",
"modules_to_save": [
"embed_tokens"
],
"peft_type": "LORA",
"r": 8,
"rank_pattern": {},
"revision": null,
"target_modules": ".*decoder.*.(k_proj|q_proj|v_proj|out_proj|fc1|fc2)",
"task_type": null
}