Skip to content

Commit 1eb7333

Browse files
committed
feat: parameterize data file and prompt file input
1 parent 364aa18 commit 1eb7333

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

azureml/eval_prompts.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
$schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
22
command: >
33
python -m autora.doc.pipelines.main eval-prompts
4-
${{inputs.data_dir}}/data.jsonl
5-
${{inputs.data_dir}}/all_prompt.json
4+
${{inputs.data_dir}}/{{data_file}}
5+
${{inputs.data_dir}}/{{prompts_file}}
66
--model-path ${{inputs.model_path}}
77
--param do_sample=${{inputs.do_sample}}
88
--param temperature=${{inputs.temperature}}
@@ -22,6 +22,8 @@ inputs:
2222
do_sample: 0
2323
top_p: 0.95
2424
top_k: 1
25+
data_file: data.jsonl
26+
prompts_file: prompt_list.json
2527
# using a curated environment doesn't work because we need additional packages
2628
environment: # azureml://registries/azureml/environments/acpt-pytorch-2.0-cuda11.7/versions/21
2729
image: mcr.microsoft.com/azureml/curated/acpt-pytorch-2.0-cuda11.7:21

0 commit comments

Comments
 (0)