ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-28 17:29:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

update reranker source code (#1082)

Browse Source
This commit is contained in:
wikkipedia
2024-03-27 21:43:48 +08:00
committed by GitHub
parent 3f892bd810
commit 00ace0b69c
4 changed files with 18 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/reranker/bge-reranker-base/Dockerfile
View File

@@ -3,8 +3,10 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
# please download the model from https://huggingface.co/BAAI/bge-reranker-base and put it in the same directory as Dockerfile
COPY ./bge-reranker-base ./bge-reranker-base
COPY app.py Dockerfile requirement.txt .
COPY requirement.txt .
RUN python3 -m pip install -r requirement.txt
RUN python3 -m pip install -r requirement.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
COPY app.py Dockerfile .
ENTRYPOINT python3 app.py