adapter_config.json

{
  "aggregation_type": "mean",
  "auto_mapping": {
    "base_model_class": "LlamaForCausalLM",
    "parent_library": "prompt.model.modeling_llama_custom"
  },
  "base_model_name_or_path": "mtgv/MobileLLaMA-1.4B-Chat",
  "inference_mode": true,
  "num_attention_heads": 16,
  "num_exits": 1,
  "num_layers": 24,
  "num_special_tokens": 3,
  "num_transformer_submodules": 1,
  "num_virtual_tokens": 3,
  "prefix_virtual_tokens": 10,
  "prompt_tuning_init": "TEXT",
  "prompt_tuning_init_text": "Next 2 word",
  "revision": null,
  "task_type": null,
  "token_dim": 2048,
  "tokenizer_kwargs": null,
  "tokenizer_name_or_path": "mtgv/MobileLLaMA-1.4B-Chat",
  "use_cache": true,
  "use_custom_lm_head": false,
  "use_prefix_tuning": false,
  "virtual_tokens_per_special_token": 1,
  "vt_attention_type": "ensemble"
}