Skip to content

Commit

Permalink
Update update_model_listing.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 Aug 21, 2023
1 parent 1cbec13 commit b1796b7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/update_model_listing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ jobs:
env:
bucket_model_listing_file_path: ml-models/model_listing/pre_trained_models.json
repo_model_listing_path: ./utils/model_uploader/model_listing/pretrained_model_listing.json
path_prefixes: "ml-models/huggingface/ ml-models/xyz ml-models/abc"
path_prefixes: "ml-models/huggingface/ ml-models/xyz/ ml-models/abc/"
# If you want to the model listing to include more folders you just need to modify path_prefixes
# as follows: "ml-models/first_folder/ ml-models/second_folder/ ml-models/third_folder/"
# (Split each folder with white space)
Expand All @@ -37,7 +37,7 @@ jobs:
path_prefixes="${{ env.path_prefixes }}"
for prefix in $path_prefixes
do
aws s3api head-object --bucket ${{ secrets.PERSONAL_MODEL_BUCKET }} --key $prefix > /dev/null 2>&1 || FOLDER_NOT_EXIST=true
aws s3 ls s3://${{ secrets.PERSONAL_MODEL_BUCKET }}/$prefix > /dev/null 2>&1 || FOLDER_NOT_EXIST=true
if [[ -z $FOLDER_NOT_EXIST ]]
then
aws s3api list-objects --bucket ${{ secrets.PERSONAL_MODEL_BUCKET }} --prefix $prefix --query "Contents[].{Key: Key}" --output text | grep "/config.json$" >> config_paths.txt
Expand Down

0 comments on commit b1796b7

Please sign in to comment.