{
  "case_name": "multitask_prompt_tuning",
  "peft_version": "0.20.0",
  "git_commit": "a5e689189fd2dd79f67ad2adee7f2e9e6328c82f",
  "torch_version": "2.13.0+cu130",
  "transformers_version": "5.15.0.dev0",
  "base_model_id": "trl-internal-testing/tiny-random-LlamaForCausalLM",
  "model_cls": "AutoModelForCausalLM",
  "inputs": {
    "input_ids": [
      [
        1,
        2,
        3
      ],
      [
        6,
        5,
        4
      ]
    ],
    "attention_mask": [
      [
        1,
        1,
        1
      ],
      [
        1,
        1,
        1
      ]
    ],
    "task_ids": [
      0,
      1
    ]
  },
  "atol": 1e-06,
  "rtol": 1e-06,
  "state_dict_keys": [
    "prefix_task_cols",
    "prefix_task_rows",
    "prompt_embeddings"
  ],
  "notes": ""
}