Skip to content

Commit

Permalink
Update model_uploader.yml
Browse files Browse the repository at this point in the history
Signed-off-by: Thanawan Atchariyachanvanit <[email protected]>
  • Loading branch information
thanawan-atc authored Jul 12, 2023
1 parent 2fec36a commit 4f5e91f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions .github/workflows/model_uploader.yml
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,7 @@ jobs:
if: ${{ github.event.inputs.model_version == 'TORCH_SCRIPT'}} || ${{ github.event.inputs.model_version == 'BOTH'}}
run: |
TORCH_FILE_PATH=$(python utils/model_uploader/save_model_file_path_to_env.py ${{ github.event.inputs.model_id }} ${{ github.event.inputs.model_version }} TORCH_SCRIPT)
echo $TORCH_FILE_PATH
aws s3api head-object --bucket opensearch-exp --key TORCH_FILE_PATH || TORCH_MODEL_NOT_EXIST=true
if [ !$TORCH_MODEL_NOT_EXIST && ${{ github.event.inputs.allow_overwrite == 'False' }} ]; then
echo "TORCH_SCRIPT Model already exists on model hub. (allow_overwrite=false)"
Expand Down

0 comments on commit 4f5e91f

Please sign in to comment.