Commit
•
8c8944c
1
Parent(s):
4e88bd5
infinity usage of reranking. Implements a cohere compatible api. (#10)
Browse files- infinity usage of reranking. Implements a cohere compatible api. (231c626908c606a13aa1f64400ca9a99e03f2ac0)
Co-authored-by: Michael <[email protected]>
README.md
CHANGED
@@ -126,6 +126,14 @@ with torch.no_grad():
|
|
126 |
# tensor([1.2315, 0.5923, 0.3041])
|
127 |
```
|
128 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
129 |
|
130 |
## Evaluation
|
131 |
|
|
|
126 |
# tensor([1.2315, 0.5923, 0.3041])
|
127 |
```
|
128 |
|
129 |
+
Usage with infinity:
|
130 |
+
|
131 |
+
[Infinity](https://github.com/michaelfeil/infinity), a MIT Licensed Inference RestAPI Server.
|
132 |
+
```
|
133 |
+
docker run --gpus all -v $PWD/data:/app/.cache -p "7997":"7997" \
|
134 |
+
michaelf34/infinity:0.0.68 \
|
135 |
+
v2 --model-id Alibaba-NLP/gte-multilingual-reranker-base --revision "main" --dtype bfloat16 --batch-size 32 --device cuda --engine torch --port 7997
|
136 |
+
```
|
137 |
|
138 |
## Evaluation
|
139 |
|