| { | |
| "case_name": "prefix_tuning", | |
| "peft_version": "0.20.0", | |
| "git_commit": "cf08deeeba784f3264dfb39817fb5c519d12e27d", | |
| "torch_version": "2.13.0+cu130", | |
| "transformers_version": "5.15.0.dev0", | |
| "base_model_id": "peft-internal-testing/tiny-random-OPTForCausalLM", | |
| "model_cls": "AutoModelForCausalLM", | |
| "inputs": { | |
| "input_ids": [ | |
| [ | |
| 1, | |
| 2, | |
| 3 | |
| ], | |
| [ | |
| 6, | |
| 5, | |
| 4 | |
| ] | |
| ], | |
| "attention_mask": [ | |
| [ | |
| 1, | |
| 1, | |
| 1 | |
| ], | |
| [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| ] | |
| }, | |
| "atol": 1e-06, | |
| "rtol": 1e-06, | |
| "state_dict_keys": [ | |
| "prompt_embeddings" | |
| ], | |
| "notes": "" | |
| } |