Skip to content

Commit b9a012c

Browse files
committed
update ci
1 parent 1ba954d commit b9a012c

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

.github/workflows/model-to-s3-from-hf-hub.yaml

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,13 @@ permissions:
1313

1414
env:
1515
AWS_REGION: "us-east-1"
16-
HF_MODEL_NAME: "tabularisai/multilingual-sentiment-analysis"
1716
ENVIRONMENT: ${{ github.ref_name == 'main' && 'prod' || github.ref_name == 'test' && 'test' }}
1817

18+
HF_USERNAME: "tabularisai"
19+
HF_REPOSITORY_NAME: "multilingual-sentiment-analysis"
20+
1921
jobs:
20-
deploy_model:
22+
copy_model_to_s3:
2123
runs-on: ubuntu-latest
2224

2325
steps:
@@ -26,11 +28,11 @@ jobs:
2628

2729
- name: Clone HuggingFace Repository
2830
run: |
29-
git clone https://huggingface.co/${{ env.HF_MODEL_NAME }}
31+
git clone https://huggingface.co/${{ env.HF_USERNAME }}/${{ env.HF_REPOSITORY_NAME }}
3032
3133
- name: Create Model Archive
3234
run: |
33-
cd ${{ env.HF_MODEL_NAME }}
35+
cd ${{ env.HF_REPOSITORY_NAME }}
3436
tar -czvf model.tar.gz *
3537
3638
- name: Configure AWS Credentials
@@ -43,6 +45,6 @@ jobs:
4345

4446
- name: Upload Model to S3
4547
env:
46-
S3_BUCKET_NAME: "dev-sagemaker-us-east-1-639269844451"
47-
S3_MODEL_PATH: "${{ env.ENVIRONMENT }}/${{ env.HF_MODEL_NAME }}/model.tar.gz"
48-
run: s3 cp model.tar.gz s3://${{ env.S3_BUCKET_NAME }}/${{ env.S3_MODEL_PATH }}
48+
S3_BUCKET_NAME: "sagemaker-us-east-1-639269844451"
49+
S3_MODEL_KEY: "${{ env.ENVIRONMENT }}/${{ env.HF_USERNAME }}/${{ env.HF_REPOSITORY_NAME }}/model.tar.gz"
50+
run: s3 cp model.tar.gz s3://${{ env.S3_BUCKET_NAME }}/${{ env.S3_MODEL_KEY }}

0 commit comments

Comments
 (0)