| { | |
| "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 | |
| } |