add prompt for text2text-generation (#290)

Author: xadupre

CHANGELOGS.rst

@@ -4,6 +4,7 @@ Change Logs
 0.8.1
 +++++
 
+* :pr:`290`: adds one prompt for text2text-generation
 * :pr:`289`: adds command line options ``--exppo`` to give the exporter additional options
 * :pr:`287`: adds input ``'inputs_prompt'`` to test a LLM, meant to be used during validation
 * :pr:`288`: add .contiguous in torch.cond branch (attention patch for sdpa implementation)

onnx_diagnostic/tasks/text2text_generation.py

@@ -151,6 +151,7 @@ def get_inputs(
         assert (
             add_second_input > 0
         ), f"Not implemented for add_second_input={add_second_input}."
+        res["inputs_prompt"] = dict(input_ids=torch.randint(1000, 30000, (1, 11)))
         res["inputs2"] = get_inputs(
             model=model,
             config=config,