michaelfeil
commited on
Commit
•
7473d37
1
Parent(s):
00d8e57
Update README.md
Browse files
README.md
CHANGED
@@ -172,6 +172,16 @@ with torch.no_grad():
|
|
172 |
print(scores)
|
173 |
```
|
174 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
175 |
## Load model in local
|
176 |
|
177 |
1. make sure `gemma_config.py` and `gemma_model.py` from [BAAI/bge-reranker-v2.5-gemma2-lightweight](https://huggingface.co/BAAI/bge-reranker-v2.5-gemma2-lightweight/tree/main) in your local path.
|
|
|
172 |
print(scores)
|
173 |
```
|
174 |
|
175 |
+
## Infinity:
|
176 |
+
|
177 |
+
For an OpenAI API-compatible local deploment and [Infinity](https://github.com/michaelfeil/infinity)
|
178 |
+
|
179 |
+
```
|
180 |
+
docker run -it --gpus all -v $volume:/app/.cache -p 7997:7997 \
|
181 |
+
michaelf34/infinity:0.0.70 \
|
182 |
+
v2 infinity_emb v2 --model-id BAAI/bge-reranker-v2.5-gemma2-lightweight --device cuda --no-bettertransformer
|
183 |
+
```
|
184 |
+
|
185 |
## Load model in local
|
186 |
|
187 |
1. make sure `gemma_config.py` and `gemma_model.py` from [BAAI/bge-reranker-v2.5-gemma2-lightweight](https://huggingface.co/BAAI/bge-reranker-v2.5-gemma2-lightweight/tree/main) in your local path.
|