-
Notifications
You must be signed in to change notification settings - Fork 1
48 lines (39 loc) · 1.63 KB
/
manual.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
name: Push to Replicate
on:
workflow_dispatch:
jobs:
build:
runs-on: [self-hosted, linux, X64]
steps:
- name: Check out code
uses: actions/checkout@v3
- name: Setup Python
uses: actions/setup-python@v4.7.1
with:
python-version: 3.11
- name: Setup Cog
uses: replicate/setup-cog@v1
with:
token: ${{ secrets.REPLICATE_API_TOKEN }}
install-cuda: 'false'
- name: Download weights
run: |
pip install -U huggingface_hub[cli]
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
with:
registry: ghcr.io
username: ninehills
password: ${{secrets.GITHUB_TOKEN}}
- name: Push to Replicate
run: |
cd bge-reranker-base/ && cog push r8.im/ninehills/bge-reranker-base
cd ../bge-reranker-large/ && cog push r8.im/ninehills/bge-reranker-large
- name: Push Image to ghcr
run: |
docker tag r8.im/ninehills/bge-reranker-base:latest ghcr.io/ninehills/bge-reranker-base:latest
docker push ghcr.io/ninehills/bge-reranker-base:latest
docker tag r8.im/ninehills/bge-reranker-large:latest ghcr.io/ninehills/bge-reranker-large:latest
docker push ghcr.io/ninehills/bge-reranker-large:latest