File size: 653 Bytes
367554b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | {
"auto_mapping": null,
"base_model_name_or_path": "trl-internal-testing/tiny-random-LlamaForCausalLM",
"inference_mode": true,
"modules_to_save": null,
"num_attention_heads": 4,
"num_layers": 2,
"num_ranks": 1,
"num_tasks": 2,
"num_transformer_submodules": 1,
"num_virtual_tokens": 10,
"peft_type": "MULTITASK_PROMPT_TUNING",
"peft_version": "0.20.0",
"prompt_tuning_init": "RANDOM",
"prompt_tuning_init_state_dict_path": null,
"prompt_tuning_init_task": 0,
"prompt_tuning_init_text": null,
"revision": null,
"task_type": "CAUSAL_LM",
"token_dim": 16,
"tokenizer_kwargs": null,
"tokenizer_name_or_path": null
} |