|
{ |
|
"auto_mapping": null, |
|
"base_model_name_or_path": "/mnt/parscratch/users/acp23ay/private/models/bloom-1b1-sw-clp", |
|
"bias": "none", |
|
"fan_in_fan_out": false, |
|
"inference_mode": true, |
|
"init_lora_weights": true, |
|
"layers_pattern": null, |
|
"layers_to_transform": null, |
|
"lora_alpha": 32, |
|
"lora_dropout": 0.05, |
|
"modules_to_save": [ |
|
"lm_head", |
|
"word_embeddings" |
|
], |
|
"peft_type": "LORA", |
|
"r": 8, |
|
"revision": null, |
|
"target_modules": [ |
|
"query_key_value", |
|
"dense", |
|
"dense_h_to_4h", |
|
"dense_4h_to_h" |
|
], |
|
"task_type": "CAUSAL_LM" |
|
} |