Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
ninehills committed Nov 21, 2023
1 parent c9e7173 commit d0b83c3
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions .github/workflows/manual.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@ jobs:
- name: Download weights
run: |
pip install -U huggingface_hub[cli]
huggingface-cli download BAAI/bge-reranker-large --local-dir bge-reranker-large/checkpoints --local-dir-use-symlink False
huggingface-cli download BAAI/bge-reranker-base --local-dir bge-reranker-base/checkpoints --local-dir-use-symlink False
huggingface-cli download BAAI/bge-reranker-large > .tmp && rsync -avL --exclude '*.safetensors' $(head -1 .tmp)/ bge-reranker-large/checkpoints
huggingface-cli download BAAI/bge-reranker-base > .tmp && rsync -avL --exclude '*.safetensors' $(head -1 .tmp)/ bge-reranker-base/checkpoints
- name: 'Login to GitHub Container Registry'
uses: docker/login-action@v1
Expand All @@ -34,12 +34,17 @@ jobs:
username: ninehills
password: ${{secrets.GITHUB_TOKEN}}

- name: Build & Push Image
- name: Build Image
run: |
cd bge-reranker-base/ && cog build -t ghcr.io/ninehills/bge-reranker-base:latest && docker push ghcr.io/ninehills/bge-reranker-base:latest
cd bge-reranker-large/ && cog build -t ghcr.io/ninehills/bge-reranker-large:latest && docker push ghcr.io/ninehills/bge-reranker-large:latest
cd bge-reranker-base/ && cog build -t ghcr.io/ninehills/bge-reranker-base:latest
cd bge-reranker-large/ && cog build -t ghcr.io/ninehills/bge-reranker-large:latest
- name: Push Image to ghcr
run: |
docker push ghcr.io/ninehills/bge-reranker-base:latest
docker push ghcr.io/ninehills/bge-reranker-large:latest
- name: Push to Replicate
run: |
cog push ninehills/bge-reranker-base
cog push ninehills/bge-reranker-large
cog push ninehills/bge-reranker-large

0 comments on commit d0b83c3

Please sign in to comment.