Sentence Similarity
sentence-transformers
Safetensors
qwen2
feature-extraction
Generated from Trainer
dataset_size:21769
loss:MultipleNegativesRankingLoss
text-embeddings-inference
Instructions to use am-azadi/KaLM-embedding-multilingual-mini-v1_Fine_Tuned_1e with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use am-azadi/KaLM-embedding-multilingual-mini-v1_Fine_Tuned_1e with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("am-azadi/KaLM-embedding-multilingual-mini-v1_Fine_Tuned_1e") sentences = [ "Blooming Canals of Venice, Italy. by: [IG] ", "This comparison shows the values of gasoline in Cádiz and in Gibraltar in 2021 The comparison of fuel prices circulates in Spain at least since 2018", "Genuine image of a Venice canal laden with lotus blossoms The lotus blossoms were digitally inserted into this image by a graphic artist", "PT deputy presented PL for police to carry unloaded weapons Bill \"5439/2022\" for police officers to carry unloaded weapons does not exist" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- .gitattributes +1 -0
- 1_Pooling/config.json +10 -0
- README.md +397 -0
- added_tokens.json +5 -0
- config.json +29 -0
- config_sentence_transformers.json +13 -0
- merges.txt +0 -0
- model.safetensors +3 -0
- modules.json +20 -0
- sentence_bert_config.json +4 -0
- special_tokens_map.json +20 -0
- tokenizer.json +3 -0
- tokenizer_config.json +57 -0
- vocab.json +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
1_Pooling/config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"word_embedding_dimension": 896,
|
| 3 |
+
"pooling_mode_cls_token": false,
|
| 4 |
+
"pooling_mode_mean_tokens": true,
|
| 5 |
+
"pooling_mode_max_tokens": false,
|
| 6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
+
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": true
|
| 10 |
+
}
|
README.md
ADDED
|
@@ -0,0 +1,397 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- sentence-transformers
|
| 4 |
+
- sentence-similarity
|
| 5 |
+
- feature-extraction
|
| 6 |
+
- generated_from_trainer
|
| 7 |
+
- dataset_size:25743
|
| 8 |
+
- loss:MultipleNegativesRankingLoss
|
| 9 |
+
base_model: HIT-TMG/KaLM-embedding-multilingual-mini-v1
|
| 10 |
+
widget:
|
| 11 |
+
- source_sentence: Why test facility of Covid-19 has a logo of Anubis, the god of
|
| 12 |
+
the dead? COVID-19 MOBILE TESTING FACILITY AF SA COVID-19 MOBILE TESTING FACILITY
|
| 13 |
+
COMID-19 COVID-19 MOBILE TESTING FACILITY P covid G
|
| 14 |
+
sentences:
|
| 15 |
+
- Le dioxyde chlore soigne le covid et "purifie" des vaccins
|
| 16 |
+
- Los camiones de pruebas de COVID-19 tienen el logo de Anubis, dios egipcio de
|
| 17 |
+
los muertos
|
| 18 |
+
- Eleitores com mais de 60 anos não precisam votar nas eleições municipais de 2020
|
| 19 |
+
devido à pandemia de covid-19
|
| 20 |
+
- source_sentence: Remember to get vaccinated or a vaccinated person might get sick
|
| 21 |
+
from the virus they got vaccinated against because you're not vaccinated.
|
| 22 |
+
sentences:
|
| 23 |
+
- Rafael Lopez Aliaga tuiteó que las mujeres casadas deben dejar su vida social
|
| 24 |
+
y dedicarse a su marido e hijos
|
| 25 |
+
- Video of Clarence Thomas ignoring two people who offered to shake his hand in
|
| 26 |
+
2021y 2021
|
| 27 |
+
- '''Remember to get vaccinated or a vaccinated person might get sick from the virus
|
| 28 |
+
they got vaccinated against because you''re not vaccinated'''
|
| 29 |
+
- source_sentence: Situação caótica obriga as autoridades a abrirem um Vala Comum.
|
| 30 |
+
. . Enquanto isso África mantém os seus casos de Infecção por Covid-19, com elevados
|
| 31 |
+
casos de Recuperação.
|
| 32 |
+
sentences:
|
| 33 |
+
- Esta foto mostra que em Nova York abrem vala comum para pessoas mortas por coronavírus
|
| 34 |
+
COVID-19.
|
| 35 |
+
- Video shows microburst over Karachi in July 2022
|
| 36 |
+
- Este venado juega en una playa de Mazatlán, durante el confinamiento por el coronavirus
|
| 37 |
+
- source_sentence: Good morning. Wish you a happy Sunday. First picture of Sunrise
|
| 38 |
+
on earth from India's Chandrayan-2.
|
| 39 |
+
sentences:
|
| 40 |
+
- First Photographs Of Earth Sent By Chandrayaan 2 Released By ISRO
|
| 41 |
+
- Video shows US protesters breaching security at the White House
|
| 42 |
+
- This is an original video of Hindu monk Swami Vivekanand's speech at a religious
|
| 43 |
+
conference in the US in September 1893.
|
| 44 |
+
- source_sentence: SAFETY DATA THE C V 243,380 surveyed 51.1% reported any adverse
|
| 45 |
+
event 22.3% reported missing work, study or routine duties 1.6% reported seeing
|
| 46 |
+
a doctor or going to emergency departmentHow can they force people to get something
|
| 47 |
+
with stats like this...
|
| 48 |
+
sentences:
|
| 49 |
+
- Covid-19 vaccine safety data from Australian vaccination monitoring agency
|
| 50 |
+
- Screenshots zeigen, dass bei der französischen Präsidentschaftswahl über 2 Millionen
|
| 51 |
+
Stimmen für Marine Le Pen verschwunden sind.
|
| 52 |
+
- Cette étude prouve-t-elle que le masque est dangereux pour la santé ?
|
| 53 |
+
pipeline_tag: sentence-similarity
|
| 54 |
+
library_name: sentence-transformers
|
| 55 |
+
---
|
| 56 |
+
|
| 57 |
+
# SentenceTransformer based on HIT-TMG/KaLM-embedding-multilingual-mini-v1
|
| 58 |
+
|
| 59 |
+
This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [HIT-TMG/KaLM-embedding-multilingual-mini-v1](https://huggingface.co/HIT-TMG/KaLM-embedding-multilingual-mini-v1). It maps sentences & paragraphs to a 896-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
| 60 |
+
|
| 61 |
+
## Model Details
|
| 62 |
+
|
| 63 |
+
### Model Description
|
| 64 |
+
- **Model Type:** Sentence Transformer
|
| 65 |
+
- **Base model:** [HIT-TMG/KaLM-embedding-multilingual-mini-v1](https://huggingface.co/HIT-TMG/KaLM-embedding-multilingual-mini-v1) <!-- at revision 685312fd77f877ad457efcf17bf31b5de0a8ed1c -->
|
| 66 |
+
- **Maximum Sequence Length:** 512 tokens
|
| 67 |
+
- **Output Dimensionality:** 896 dimensions
|
| 68 |
+
- **Similarity Function:** Cosine Similarity
|
| 69 |
+
<!-- - **Training Dataset:** Unknown -->
|
| 70 |
+
<!-- - **Language:** Unknown -->
|
| 71 |
+
<!-- - **License:** Unknown -->
|
| 72 |
+
|
| 73 |
+
### Model Sources
|
| 74 |
+
|
| 75 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| 76 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
| 77 |
+
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
| 78 |
+
|
| 79 |
+
### Full Model Architecture
|
| 80 |
+
|
| 81 |
+
```
|
| 82 |
+
SentenceTransformer(
|
| 83 |
+
(0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: Qwen2Model
|
| 84 |
+
(1): Pooling({'word_embedding_dimension': 896, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
|
| 85 |
+
(2): Normalize()
|
| 86 |
+
)
|
| 87 |
+
```
|
| 88 |
+
|
| 89 |
+
## Usage
|
| 90 |
+
|
| 91 |
+
### Direct Usage (Sentence Transformers)
|
| 92 |
+
|
| 93 |
+
First install the Sentence Transformers library:
|
| 94 |
+
|
| 95 |
+
```bash
|
| 96 |
+
pip install -U sentence-transformers
|
| 97 |
+
```
|
| 98 |
+
|
| 99 |
+
Then you can load this model and run inference.
|
| 100 |
+
```python
|
| 101 |
+
from sentence_transformers import SentenceTransformer
|
| 102 |
+
|
| 103 |
+
# Download from the 🤗 Hub
|
| 104 |
+
model = SentenceTransformer("sentence_transformers_model_id")
|
| 105 |
+
# Run inference
|
| 106 |
+
sentences = [
|
| 107 |
+
'SAFETY DATA THE C V 243,380 surveyed 51.1% reported any adverse event 22.3% reported missing work, study or routine duties 1.6% reported seeing a doctor or going to emergency departmentHow can they force people to get something with stats like this...',
|
| 108 |
+
'Covid-19 vaccine safety data from Australian vaccination monitoring agency',
|
| 109 |
+
'Screenshots zeigen, dass bei der französischen Präsidentschaftswahl über 2 Millionen Stimmen für Marine Le Pen verschwunden sind.',
|
| 110 |
+
]
|
| 111 |
+
embeddings = model.encode(sentences)
|
| 112 |
+
print(embeddings.shape)
|
| 113 |
+
# [3, 896]
|
| 114 |
+
|
| 115 |
+
# Get the similarity scores for the embeddings
|
| 116 |
+
similarities = model.similarity(embeddings, embeddings)
|
| 117 |
+
print(similarities.shape)
|
| 118 |
+
# [3, 3]
|
| 119 |
+
```
|
| 120 |
+
|
| 121 |
+
<!--
|
| 122 |
+
### Direct Usage (Transformers)
|
| 123 |
+
|
| 124 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
| 125 |
+
|
| 126 |
+
</details>
|
| 127 |
+
-->
|
| 128 |
+
|
| 129 |
+
<!--
|
| 130 |
+
### Downstream Usage (Sentence Transformers)
|
| 131 |
+
|
| 132 |
+
You can finetune this model on your own dataset.
|
| 133 |
+
|
| 134 |
+
<details><summary>Click to expand</summary>
|
| 135 |
+
|
| 136 |
+
</details>
|
| 137 |
+
-->
|
| 138 |
+
|
| 139 |
+
<!--
|
| 140 |
+
### Out-of-Scope Use
|
| 141 |
+
|
| 142 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 143 |
+
-->
|
| 144 |
+
|
| 145 |
+
<!--
|
| 146 |
+
## Bias, Risks and Limitations
|
| 147 |
+
|
| 148 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 149 |
+
-->
|
| 150 |
+
|
| 151 |
+
<!--
|
| 152 |
+
### Recommendations
|
| 153 |
+
|
| 154 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 155 |
+
-->
|
| 156 |
+
|
| 157 |
+
## Training Details
|
| 158 |
+
|
| 159 |
+
### Training Dataset
|
| 160 |
+
|
| 161 |
+
#### Unnamed Dataset
|
| 162 |
+
|
| 163 |
+
* Size: 25,743 training samples
|
| 164 |
+
* Columns: <code>sentence_0</code>, <code>sentence_1</code>, and <code>label</code>
|
| 165 |
+
* Approximate statistics based on the first 1000 samples:
|
| 166 |
+
| | sentence_0 | sentence_1 | label |
|
| 167 |
+
|:--------|:-----------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|:--------------------------------------------------------------|
|
| 168 |
+
| type | string | string | float |
|
| 169 |
+
| details | <ul><li>min: 0 tokens</li><li>mean: 135.8 tokens</li><li>max: 512 tokens</li></ul> | <ul><li>min: 4 tokens</li><li>mean: 20.43 tokens</li><li>max: 150 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
|
| 170 |
+
* Samples:
|
| 171 |
+
| sentence_0 | sentence_1 | label |
|
| 172 |
+
|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------------|:-----------------|
|
| 173 |
+
| <code>Vía Los Patios Pamplona Santander asesinan dos miembros de la policía Nacional, Noticia en desarrollo. Porqué los medios no reseña éstas noticias? Hay orden de quién y para que ocultarlo? Seguimos investigando....</code> | <code>Asesinan a dos policías en la vía Los Patios-Pamplona, Norte de Santander</code> | <code>1.0</code> |
|
| 174 |
+
| <code>"As a UNICEF ambassador, I cannot play against people who kill innocent palestinian children. We had to cancel the game because we are humans before footballers." -Lionel MessiLionel Messi is refusing to play friendly match between Argentina and Israel you don't need to be muslim to stand up for Palestine,u just need to a human!</code> | <code>Messi boycotts Israel match for children killed in Palestine</code> | <code>1.0</code> |
|
| 175 |
+
| <code>KEBERHASILAN PRANCIS MENJADI TERORIS Foto ini dibuat pada tahun 1955 di tengah penjajahan Prancis atas Kongo. Dalam foto tersebut seorang Ayah membawa seorang anak Afrika untuk anak-anaknya sebagai "hiburan." Perlu diketahui, bahwa Prancis berhasil membunuh 10 hingga 15 juta penduduk Kongo dalam waktu 50 tahun penjajahannya. Prancis juga berhasil memotong ribuan tangan anak-anak di perkebunan karet dan lahan lainnya sebagai bentuk hukuman atas kegagalan sang Ayah dalam mengumpulkan jumlah karet ataupun bahan tambang lainnya. Sampai akhirnya Negara Kongo dinamakan: "Negara Tangan Yang Terpotong." صورة من سنة ١٩٥٥ أثناء احتلال الفرنسي للكونغو حين أتى أب بطفل أفريقي لأبناءه للتسلية به.. حيث قتلت فرنسا في الكونغو لوحدها مابين ١٠ و١٥ مليون كونغولي في خمسين سنة من استعمارها. وقطعت أيدي آلاف من الأطفال في حقول المطاط وغيرها عقاباً لأي أب كونغولي لا ينجح في جمع الكمية المطلوبة من المطاط أو المعادن، حتى سميت الكونغو، "بلد الأيدي المقطوعة". إرهاب فرنسا ..</code> | <code>Foto anak-anak saat Kongo dijajah Prancis</code> | <code>1.0</code> |
|
| 176 |
+
* Loss: [<code>MultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#multiplenegativesrankingloss) with these parameters:
|
| 177 |
+
```json
|
| 178 |
+
{
|
| 179 |
+
"scale": 20.0,
|
| 180 |
+
"similarity_fct": "cos_sim"
|
| 181 |
+
}
|
| 182 |
+
```
|
| 183 |
+
|
| 184 |
+
### Training Hyperparameters
|
| 185 |
+
#### Non-Default Hyperparameters
|
| 186 |
+
|
| 187 |
+
- `per_device_train_batch_size`: 2
|
| 188 |
+
- `per_device_eval_batch_size`: 2
|
| 189 |
+
- `num_train_epochs`: 1
|
| 190 |
+
- `multi_dataset_batch_sampler`: round_robin
|
| 191 |
+
|
| 192 |
+
#### All Hyperparameters
|
| 193 |
+
<details><summary>Click to expand</summary>
|
| 194 |
+
|
| 195 |
+
- `overwrite_output_dir`: False
|
| 196 |
+
- `do_predict`: False
|
| 197 |
+
- `eval_strategy`: no
|
| 198 |
+
- `prediction_loss_only`: True
|
| 199 |
+
- `per_device_train_batch_size`: 2
|
| 200 |
+
- `per_device_eval_batch_size`: 2
|
| 201 |
+
- `per_gpu_train_batch_size`: None
|
| 202 |
+
- `per_gpu_eval_batch_size`: None
|
| 203 |
+
- `gradient_accumulation_steps`: 1
|
| 204 |
+
- `eval_accumulation_steps`: None
|
| 205 |
+
- `torch_empty_cache_steps`: None
|
| 206 |
+
- `learning_rate`: 5e-05
|
| 207 |
+
- `weight_decay`: 0.0
|
| 208 |
+
- `adam_beta1`: 0.9
|
| 209 |
+
- `adam_beta2`: 0.999
|
| 210 |
+
- `adam_epsilon`: 1e-08
|
| 211 |
+
- `max_grad_norm`: 1
|
| 212 |
+
- `num_train_epochs`: 1
|
| 213 |
+
- `max_steps`: -1
|
| 214 |
+
- `lr_scheduler_type`: linear
|
| 215 |
+
- `lr_scheduler_kwargs`: {}
|
| 216 |
+
- `warmup_ratio`: 0.0
|
| 217 |
+
- `warmup_steps`: 0
|
| 218 |
+
- `log_level`: passive
|
| 219 |
+
- `log_level_replica`: warning
|
| 220 |
+
- `log_on_each_node`: True
|
| 221 |
+
- `logging_nan_inf_filter`: True
|
| 222 |
+
- `save_safetensors`: True
|
| 223 |
+
- `save_on_each_node`: False
|
| 224 |
+
- `save_only_model`: False
|
| 225 |
+
- `restore_callback_states_from_checkpoint`: False
|
| 226 |
+
- `no_cuda`: False
|
| 227 |
+
- `use_cpu`: False
|
| 228 |
+
- `use_mps_device`: False
|
| 229 |
+
- `seed`: 42
|
| 230 |
+
- `data_seed`: None
|
| 231 |
+
- `jit_mode_eval`: False
|
| 232 |
+
- `use_ipex`: False
|
| 233 |
+
- `bf16`: False
|
| 234 |
+
- `fp16`: False
|
| 235 |
+
- `fp16_opt_level`: O1
|
| 236 |
+
- `half_precision_backend`: auto
|
| 237 |
+
- `bf16_full_eval`: False
|
| 238 |
+
- `fp16_full_eval`: False
|
| 239 |
+
- `tf32`: None
|
| 240 |
+
- `local_rank`: 0
|
| 241 |
+
- `ddp_backend`: None
|
| 242 |
+
- `tpu_num_cores`: None
|
| 243 |
+
- `tpu_metrics_debug`: False
|
| 244 |
+
- `debug`: []
|
| 245 |
+
- `dataloader_drop_last`: False
|
| 246 |
+
- `dataloader_num_workers`: 0
|
| 247 |
+
- `dataloader_prefetch_factor`: None
|
| 248 |
+
- `past_index`: -1
|
| 249 |
+
- `disable_tqdm`: False
|
| 250 |
+
- `remove_unused_columns`: True
|
| 251 |
+
- `label_names`: None
|
| 252 |
+
- `load_best_model_at_end`: False
|
| 253 |
+
- `ignore_data_skip`: False
|
| 254 |
+
- `fsdp`: []
|
| 255 |
+
- `fsdp_min_num_params`: 0
|
| 256 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
| 257 |
+
- `fsdp_transformer_layer_cls_to_wrap`: None
|
| 258 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
| 259 |
+
- `deepspeed`: None
|
| 260 |
+
- `label_smoothing_factor`: 0.0
|
| 261 |
+
- `optim`: adamw_torch
|
| 262 |
+
- `optim_args`: None
|
| 263 |
+
- `adafactor`: False
|
| 264 |
+
- `group_by_length`: False
|
| 265 |
+
- `length_column_name`: length
|
| 266 |
+
- `ddp_find_unused_parameters`: None
|
| 267 |
+
- `ddp_bucket_cap_mb`: None
|
| 268 |
+
- `ddp_broadcast_buffers`: False
|
| 269 |
+
- `dataloader_pin_memory`: True
|
| 270 |
+
- `dataloader_persistent_workers`: False
|
| 271 |
+
- `skip_memory_metrics`: True
|
| 272 |
+
- `use_legacy_prediction_loop`: False
|
| 273 |
+
- `push_to_hub`: False
|
| 274 |
+
- `resume_from_checkpoint`: None
|
| 275 |
+
- `hub_model_id`: None
|
| 276 |
+
- `hub_strategy`: every_save
|
| 277 |
+
- `hub_private_repo`: None
|
| 278 |
+
- `hub_always_push`: False
|
| 279 |
+
- `gradient_checkpointing`: False
|
| 280 |
+
- `gradient_checkpointing_kwargs`: None
|
| 281 |
+
- `include_inputs_for_metrics`: False
|
| 282 |
+
- `include_for_metrics`: []
|
| 283 |
+
- `eval_do_concat_batches`: True
|
| 284 |
+
- `fp16_backend`: auto
|
| 285 |
+
- `push_to_hub_model_id`: None
|
| 286 |
+
- `push_to_hub_organization`: None
|
| 287 |
+
- `mp_parameters`:
|
| 288 |
+
- `auto_find_batch_size`: False
|
| 289 |
+
- `full_determinism`: False
|
| 290 |
+
- `torchdynamo`: None
|
| 291 |
+
- `ray_scope`: last
|
| 292 |
+
- `ddp_timeout`: 1800
|
| 293 |
+
- `torch_compile`: False
|
| 294 |
+
- `torch_compile_backend`: None
|
| 295 |
+
- `torch_compile_mode`: None
|
| 296 |
+
- `dispatch_batches`: None
|
| 297 |
+
- `split_batches`: None
|
| 298 |
+
- `include_tokens_per_second`: False
|
| 299 |
+
- `include_num_input_tokens_seen`: False
|
| 300 |
+
- `neftune_noise_alpha`: None
|
| 301 |
+
- `optim_target_modules`: None
|
| 302 |
+
- `batch_eval_metrics`: False
|
| 303 |
+
- `eval_on_start`: False
|
| 304 |
+
- `use_liger_kernel`: False
|
| 305 |
+
- `eval_use_gather_object`: False
|
| 306 |
+
- `average_tokens_across_devices`: False
|
| 307 |
+
- `prompts`: None
|
| 308 |
+
- `batch_sampler`: batch_sampler
|
| 309 |
+
- `multi_dataset_batch_sampler`: round_robin
|
| 310 |
+
|
| 311 |
+
</details>
|
| 312 |
+
|
| 313 |
+
### Training Logs
|
| 314 |
+
| Epoch | Step | Training Loss |
|
| 315 |
+
|:------:|:-----:|:-------------:|
|
| 316 |
+
| 0.0388 | 500 | 0.0311 |
|
| 317 |
+
| 0.0777 | 1000 | 0.0728 |
|
| 318 |
+
| 0.1165 | 1500 | 0.0703 |
|
| 319 |
+
| 0.1554 | 2000 | 0.0463 |
|
| 320 |
+
| 0.1942 | 2500 | 0.0541 |
|
| 321 |
+
| 0.2331 | 3000 | 0.0422 |
|
| 322 |
+
| 0.2719 | 3500 | 0.0523 |
|
| 323 |
+
| 0.3108 | 4000 | 0.035 |
|
| 324 |
+
| 0.3496 | 4500 | 0.0671 |
|
| 325 |
+
| 0.3884 | 5000 | 0.0343 |
|
| 326 |
+
| 0.4273 | 5500 | 0.0438 |
|
| 327 |
+
| 0.4661 | 6000 | 0.0417 |
|
| 328 |
+
| 0.5050 | 6500 | 0.0502 |
|
| 329 |
+
| 0.5438 | 7000 | 0.0304 |
|
| 330 |
+
| 0.5827 | 7500 | 0.0185 |
|
| 331 |
+
| 0.6215 | 8000 | 0.0251 |
|
| 332 |
+
| 0.6603 | 8500 | 0.0186 |
|
| 333 |
+
| 0.6992 | 9000 | 0.0413 |
|
| 334 |
+
| 0.7380 | 9500 | 0.0229 |
|
| 335 |
+
| 0.7769 | 10000 | 0.0438 |
|
| 336 |
+
| 0.8157 | 10500 | 0.0245 |
|
| 337 |
+
| 0.8546 | 11000 | 0.0223 |
|
| 338 |
+
| 0.8934 | 11500 | 0.0328 |
|
| 339 |
+
| 0.9323 | 12000 | 0.0144 |
|
| 340 |
+
| 0.9711 | 12500 | 0.0215 |
|
| 341 |
+
|
| 342 |
+
|
| 343 |
+
### Framework Versions
|
| 344 |
+
- Python: 3.11.11
|
| 345 |
+
- Sentence Transformers: 3.4.1
|
| 346 |
+
- Transformers: 4.48.3
|
| 347 |
+
- PyTorch: 2.5.1+cu124
|
| 348 |
+
- Accelerate: 1.3.0
|
| 349 |
+
- Datasets: 3.3.2
|
| 350 |
+
- Tokenizers: 0.21.0
|
| 351 |
+
|
| 352 |
+
## Citation
|
| 353 |
+
|
| 354 |
+
### BibTeX
|
| 355 |
+
|
| 356 |
+
#### Sentence Transformers
|
| 357 |
+
```bibtex
|
| 358 |
+
@inproceedings{reimers-2019-sentence-bert,
|
| 359 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
| 360 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
| 361 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
| 362 |
+
month = "11",
|
| 363 |
+
year = "2019",
|
| 364 |
+
publisher = "Association for Computational Linguistics",
|
| 365 |
+
url = "https://arxiv.org/abs/1908.10084",
|
| 366 |
+
}
|
| 367 |
+
```
|
| 368 |
+
|
| 369 |
+
#### MultipleNegativesRankingLoss
|
| 370 |
+
```bibtex
|
| 371 |
+
@misc{henderson2017efficient,
|
| 372 |
+
title={Efficient Natural Language Response Suggestion for Smart Reply},
|
| 373 |
+
author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
|
| 374 |
+
year={2017},
|
| 375 |
+
eprint={1705.00652},
|
| 376 |
+
archivePrefix={arXiv},
|
| 377 |
+
primaryClass={cs.CL}
|
| 378 |
+
}
|
| 379 |
+
```
|
| 380 |
+
|
| 381 |
+
<!--
|
| 382 |
+
## Glossary
|
| 383 |
+
|
| 384 |
+
*Clearly define terms in order to be accessible across audiences.*
|
| 385 |
+
-->
|
| 386 |
+
|
| 387 |
+
<!--
|
| 388 |
+
## Model Card Authors
|
| 389 |
+
|
| 390 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| 391 |
+
-->
|
| 392 |
+
|
| 393 |
+
<!--
|
| 394 |
+
## Model Card Contact
|
| 395 |
+
|
| 396 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| 397 |
+
-->
|
added_tokens.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"<|endoftext|>": 151643,
|
| 3 |
+
"<|im_end|>": 151645,
|
| 4 |
+
"<|im_start|>": 151644
|
| 5 |
+
}
|
config.json
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "HIT-TMG/KaLM-embedding-multilingual-mini-v1",
|
| 3 |
+
"architectures": [
|
| 4 |
+
"Qwen2Model"
|
| 5 |
+
],
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"bos_token_id": 151643,
|
| 8 |
+
"eos_token_id": 151643,
|
| 9 |
+
"hidden_act": "silu",
|
| 10 |
+
"hidden_size": 896,
|
| 11 |
+
"initializer_range": 0.02,
|
| 12 |
+
"intermediate_size": 4864,
|
| 13 |
+
"max_position_embeddings": 131072,
|
| 14 |
+
"max_window_layers": 24,
|
| 15 |
+
"model_type": "qwen2",
|
| 16 |
+
"num_attention_heads": 14,
|
| 17 |
+
"num_hidden_layers": 24,
|
| 18 |
+
"num_key_value_heads": 2,
|
| 19 |
+
"rms_norm_eps": 1e-06,
|
| 20 |
+
"rope_scaling": null,
|
| 21 |
+
"rope_theta": 1000000.0,
|
| 22 |
+
"sliding_window": null,
|
| 23 |
+
"tie_word_embeddings": true,
|
| 24 |
+
"torch_dtype": "float32",
|
| 25 |
+
"transformers_version": "4.48.3",
|
| 26 |
+
"use_cache": false,
|
| 27 |
+
"use_sliding_window": false,
|
| 28 |
+
"vocab_size": 151936
|
| 29 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"__version__": {
|
| 3 |
+
"sentence_transformers": "3.4.1",
|
| 4 |
+
"transformers": "4.48.3",
|
| 5 |
+
"pytorch": "2.5.1+cu124"
|
| 6 |
+
},
|
| 7 |
+
"prompts": {
|
| 8 |
+
"query": "",
|
| 9 |
+
"document": ""
|
| 10 |
+
},
|
| 11 |
+
"default_prompt_name": null,
|
| 12 |
+
"similarity_fn_name": "cosine"
|
| 13 |
+
}
|
merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e3986085707cebe4569391fd69be33de20060d453c527e9884f901fca55be4bf
|
| 3 |
+
size 1976161736
|
modules.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"idx": 0,
|
| 4 |
+
"name": "0",
|
| 5 |
+
"path": "",
|
| 6 |
+
"type": "sentence_transformers.models.Transformer"
|
| 7 |
+
},
|
| 8 |
+
{
|
| 9 |
+
"idx": 1,
|
| 10 |
+
"name": "1",
|
| 11 |
+
"path": "1_Pooling",
|
| 12 |
+
"type": "sentence_transformers.models.Pooling"
|
| 13 |
+
},
|
| 14 |
+
{
|
| 15 |
+
"idx": 2,
|
| 16 |
+
"name": "2",
|
| 17 |
+
"path": "2_Normalize",
|
| 18 |
+
"type": "sentence_transformers.models.Normalize"
|
| 19 |
+
}
|
| 20 |
+
]
|
sentence_bert_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"max_seq_length": 512,
|
| 3 |
+
"do_lower_case": false
|
| 4 |
+
}
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
"<|im_start|>",
|
| 4 |
+
"<|im_end|>"
|
| 5 |
+
],
|
| 6 |
+
"eos_token": {
|
| 7 |
+
"content": "<|endoftext|>",
|
| 8 |
+
"lstrip": false,
|
| 9 |
+
"normalized": false,
|
| 10 |
+
"rstrip": false,
|
| 11 |
+
"single_word": false
|
| 12 |
+
},
|
| 13 |
+
"pad_token": {
|
| 14 |
+
"content": "<|endoftext|>",
|
| 15 |
+
"lstrip": false,
|
| 16 |
+
"normalized": false,
|
| 17 |
+
"rstrip": false,
|
| 18 |
+
"single_word": false
|
| 19 |
+
}
|
| 20 |
+
}
|
tokenizer.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c8eab352ec6bb19236f8684bf40e504e86876bd4f2f43982b0561b2f07702666
|
| 3 |
+
size 11418805
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"added_tokens_decoder": {
|
| 4 |
+
"151643": {
|
| 5 |
+
"content": "<|endoftext|>",
|
| 6 |
+
"lstrip": false,
|
| 7 |
+
"normalized": false,
|
| 8 |
+
"rstrip": false,
|
| 9 |
+
"single_word": false,
|
| 10 |
+
"special": true
|
| 11 |
+
},
|
| 12 |
+
"151644": {
|
| 13 |
+
"content": "<|im_start|>",
|
| 14 |
+
"lstrip": false,
|
| 15 |
+
"normalized": false,
|
| 16 |
+
"rstrip": false,
|
| 17 |
+
"single_word": false,
|
| 18 |
+
"special": true
|
| 19 |
+
},
|
| 20 |
+
"151645": {
|
| 21 |
+
"content": "<|im_end|>",
|
| 22 |
+
"lstrip": false,
|
| 23 |
+
"normalized": false,
|
| 24 |
+
"rstrip": false,
|
| 25 |
+
"single_word": false,
|
| 26 |
+
"special": true
|
| 27 |
+
}
|
| 28 |
+
},
|
| 29 |
+
"additional_special_tokens": [
|
| 30 |
+
"<|im_start|>",
|
| 31 |
+
"<|im_end|>"
|
| 32 |
+
],
|
| 33 |
+
"auto_map": {
|
| 34 |
+
"AutoTokenizer": [
|
| 35 |
+
"HIT-TMG/KaLM-embedding-multilingual-mini-v1--tokenization_qwen.Qwen2Tokenizer",
|
| 36 |
+
"HIT-TMG/KaLM-embedding-multilingual-mini-v1--tokenization_qwen.Qwen2TokenizerFast"
|
| 37 |
+
]
|
| 38 |
+
},
|
| 39 |
+
"bos_token": null,
|
| 40 |
+
"chat_template": "{% for message in messages %}{% if loop.first and messages[0]['role'] != 'system' %}{{ '<|im_start|>system\nYou are a helpful assistant<|im_end|>\n' }}{% endif %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
|
| 41 |
+
"clean_up_tokenization_spaces": false,
|
| 42 |
+
"eos_token": "<|endoftext|>",
|
| 43 |
+
"errors": "replace",
|
| 44 |
+
"extra_special_tokens": {},
|
| 45 |
+
"max_length": 512,
|
| 46 |
+
"model_max_length": 512,
|
| 47 |
+
"pad_to_multiple_of": null,
|
| 48 |
+
"pad_token": "<|endoftext|>",
|
| 49 |
+
"pad_token_type_id": 0,
|
| 50 |
+
"padding_side": "left",
|
| 51 |
+
"split_special_tokens": false,
|
| 52 |
+
"stride": 0,
|
| 53 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
| 54 |
+
"truncation_side": "right",
|
| 55 |
+
"truncation_strategy": "longest_first",
|
| 56 |
+
"unk_token": null
|
| 57 |
+
}
|
vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|