Update README.md
Browse files
README.md
CHANGED
@@ -15,7 +15,7 @@ library_name: PyLate
|
|
15 |
metrics:
|
16 |
- accuracy
|
17 |
model-index:
|
18 |
-
- name: PyLate model based on EuroBERT/EuroBERT-
|
19 |
results:
|
20 |
- task:
|
21 |
type: col-berttriplet
|
@@ -32,10 +32,11 @@ language:
|
|
32 |
- es
|
33 |
- en
|
34 |
---
|
|
|
35 |
|
36 |
## Fine-Tuned Model
|
37 |
|
38 |
-
**`
|
39 |
|
40 |
## Base Model
|
41 |
**`EuroBERT/EuroBERT-610m`**
|
@@ -71,7 +72,7 @@ import torch
|
|
71 |
from pylate import models
|
72 |
|
73 |
# Load the ColBERT model
|
74 |
-
model = models.ColBERT("
|
75 |
|
76 |
# Move the model to GPU if available, otherwise use CPU
|
77 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
@@ -143,5 +144,7 @@ print(f"Similarity with negative document: {negative_score.item():.4f}")
|
|
143 |
This tuned model is designed for **Spanish applications** that require the use of **efficient semantic search** comparing embeddings at the token level with its MaxSim operation, ideal for **question-answering and document retrieval**.
|
144 |
|
145 |
|
146 |
-
- **Developed by:**
|
147 |
-
- **License:** apache-2.0
|
|
|
|
|
|
15 |
metrics:
|
16 |
- accuracy
|
17 |
model-index:
|
18 |
+
- name: PyLate model based on EuroBERT/EuroBERT-210m
|
19 |
results:
|
20 |
- task:
|
21 |
type: col-berttriplet
|
|
|
32 |
- es
|
33 |
- en
|
34 |
---
|
35 |
+
[<img src="https://cdn-avatars.huggingface.co/v1/production/uploads/67b2f4e49edebc815a3a4739/R1g957j1aBbx8lhZbWmxw.jpeg" width="200"/>](https://huggingface.co/fjmgAI)
|
36 |
|
37 |
## Fine-Tuned Model
|
38 |
|
39 |
+
**`fjmgAI/col1-610M-EuroBERT`**
|
40 |
|
41 |
## Base Model
|
42 |
**`EuroBERT/EuroBERT-610m`**
|
|
|
72 |
from pylate import models
|
73 |
|
74 |
# Load the ColBERT model
|
75 |
+
model = models.ColBERT("fjmgAI/col1-610M-EuroBERT", trust_remote_code=True)
|
76 |
|
77 |
# Move the model to GPU if available, otherwise use CPU
|
78 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
144 |
This tuned model is designed for **Spanish applications** that require the use of **efficient semantic search** comparing embeddings at the token level with its MaxSim operation, ideal for **question-answering and document retrieval**.
|
145 |
|
146 |
|
147 |
+
- **Developed by:** fjmgAI
|
148 |
+
- **License:** apache-2.0
|
149 |
+
|
150 |
+
[<img src="https://github.com/lightonai/pylate/blob/main/docs/img/logo.png?raw=true" width="200"/>](https://github.com/lightonai/pylate)
|