diff --git a/.github/workflows/manual.yaml b/.github/workflows/manual.yaml index 4f357a5..8eab736 100644 --- a/.github/workflows/manual.yaml +++ b/.github/workflows/manual.yaml @@ -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 @@ -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 \ No newline at end of file + cog push ninehills/bge-reranker-large