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