Skip to content

Commit 38e29dc

Browse files
author
Shrey Modi
committed
Update rollout.yml to support completion_params
1 parent 8560c88 commit 38e29dc

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.github/workflows/rollout.yml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,10 @@ on:
1717
description: 'Base URL for the model API'
1818
required: true
1919
type: string
20+
completion_params:
21+
description: 'JSON completion params (optional, includes model_kwargs)'
22+
required: false
23+
type: string
2024

2125
jobs:
2226
rollout:
@@ -43,4 +47,5 @@ jobs:
4347
python tests/github_actions/rollout_worker.py \
4448
--model "${{ inputs.model }}" \
4549
--metadata '${{ inputs.metadata }}' \
46-
--model-base-url "${{ inputs.model_base_url }}"
50+
--model-base-url "${{ inputs.model_base_url }}" \
51+
${{ inputs.completion_params && format('--completion-params ''{0}''', inputs.completion_params) || '' }}

0 commit comments

Comments
 (0)