mirror of
https://github.com/labring/FastGPT.git
synced 2025-07-28 09:03:53 +00:00
add reranker (#679)
This commit is contained in:
10
python/reranker/bge-reranker-base/Dockerfile
Normal file
10
python/reranker/bge-reranker-base/Dockerfile
Normal file
@@ -0,0 +1,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 .
|
||||
|
||||
RUN python3 -m pip install -r requirement.txt
|
||||
|
||||
ENTRYPOINT python3 app.py
|
Reference in New Issue
Block a user