Update README.md to call out GPU support #147
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: AWS Model Builder | |
| on: | |
| push: | |
| branches: | |
| - main | |
| pull_request: | |
| types: | |
| - opened | |
| - synchronize | |
| - reopened | |
| - ready_for_review | |
| jobs: | |
| build: | |
| runs-on: ubuntu-latest | |
| if: github.event.pull_request.draft == false | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| ref: ${{github.event.pull_request.head.sha || github.sha}} | |
| - name: Docker Build Task | |
| run: docker build -t segment-build . -f Dockerfile-build | |
| - name: Configure AWS Credentials | |
| uses: aws-actions/configure-aws-credentials@v1 | |
| with: | |
| aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }} | |
| aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }} | |
| aws-region: ${{ secrets.AWS_REGION }} | |
| - name: Run Container and Copy .mar Files to S3 | |
| run: | | |
| GITSHA=${{github.event.pull_request.head.sha || github.sha}} | |
| docker run \ | |
| -e AWS_SECRET_ACCESS_KEY=${{secrets.AWS_SECRET_ACCESS_KEY}} \ | |
| -e AWS_ACCESS_KEY_ID=${{secrets.AWS_ACCESS_KEY_ID}} \ | |
| -e GITSHA=$GITSHA \ | |
| segment-build \ | |
| bash -c "\ | |
| aws configure set aws_access_key_id \$AWS_ACCESS_KEY_ID && \ | |
| aws configure set aws_secret_access_key \$AWS_SECRET_ACCESS_KEY && \ | |
| aws configure set region us-east-1 && \ | |
| aws s3 cp /home/model-store/sam_vit_h_decode.mar s3://segment-anything-services-prod-685980093638-us-east-1/\${GITSHA}/ && \ | |
| aws s3 cp /home/model-store/sam_vit_h_encode.mar s3://segment-anything-services-prod-685980093638-us-east-1/\${GITSHA}/" | |