leafxyz commited on
Commit
e2c4d24
·
verified ·
1 Parent(s): e0a6e35

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -83,3 +83,4 @@ checkpoint-9400/tokenizer.json filter=lfs diff=lfs merge=lfs -text
83
  checkpoint-9600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
84
  checkpoint-9800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
85
  checkpoint-10000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
83
  checkpoint-9600/tokenizer.json filter=lfs diff=lfs merge=lfs -text
84
  checkpoint-9800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
85
  checkpoint-10000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
86
+ checkpoint-10200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-10200/1_Pooling/config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "embedding_dimension": 1024,
3
+ "pooling_mode": "lasttoken",
4
+ "include_prompt": true
5
+ }
checkpoint-10200/README.md ADDED
@@ -0,0 +1,475 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - generated_from_trainer
7
+ - dataset_size:342061
8
+ - loss:CachedMultipleNegativesRankingLoss
9
+ base_model: prestoai/qwen3-embedding-0.6b-arabic-ecom
10
+ widget:
11
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
12
+ product that best matches it
13
+
14
+ Query: نسونكس Spray'
15
+ sentences:
16
+ - Nasonex - Nasal Spray
17
+ - كابل شحن مايكرو XKIN - 2.4A
18
+ - حلوى الشوكولاتة - Choco Lapki
19
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
20
+ product that best matches it
21
+
22
+ Query: مكرونة رقم 42'
23
+ sentences:
24
+ - بنطلون رجالي - 0112
25
+ - حقيبة حزام خصر - 4862
26
+ - مكرونة الجيد معكوفة رقم 42 - 500 غ
27
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
28
+ product that best matches it
29
+
30
+ Query: سباغيتي'
31
+ sentences:
32
+ - ملعب كرة قدم - DD18
33
+ - مكرونة معكوفة - Favelli
34
+ - مكرونة سباغيتي - Favelli
35
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
36
+ product that best matches it
37
+
38
+ Query: جبنة هواء'
39
+ sentences:
40
+ - جبنة - ابو الولد
41
+ - جبنة - Hawaa
42
+ - كاني طعام كلاب البالغين دجاج - 3 ك
43
+ - source_sentence: 'Instruct: Given an Arabic e-commerce search query, retrieve the
44
+ product that best matches it
45
+
46
+ Query: شاحن تايب سي للسيارة'
47
+ sentences:
48
+ - شاحن سيارة قرين ليون مدخلين 36 وات مع كابل تايب سي - CBK
49
+ - صوص المكرونة هاينز - 365 غ
50
+ - بسكويت جولون بدون سكر شكلاتة ساندوتش
51
+ pipeline_tag: sentence-similarity
52
+ library_name: sentence-transformers
53
+ ---
54
+
55
+ # SentenceTransformer based on prestoai/qwen3-embedding-0.6b-arabic-ecom
56
+
57
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [prestoai/qwen3-embedding-0.6b-arabic-ecom](https://huggingface.co/prestoai/qwen3-embedding-0.6b-arabic-ecom) on the pairs_with_negatives and positives datasets. It maps sentences & paragraphs to a 1024-dimensional dense vector space and can be used for retrieval.
58
+
59
+ ## Model Details
60
+
61
+ ### Model Description
62
+ - **Model Type:** Sentence Transformer
63
+ - **Base model:** [prestoai/qwen3-embedding-0.6b-arabic-ecom](https://huggingface.co/prestoai/qwen3-embedding-0.6b-arabic-ecom) <!-- at revision 80f273fd53c6644d65e14a2ac1fbf74b8c924097 -->
64
+ - **Maximum Sequence Length:** 128 tokens
65
+ - **Output Dimensionality:** 1024 dimensions
66
+ - **Similarity Function:** Cosine Similarity
67
+ - **Supported Modality:** Text
68
+ - **Training Datasets:**
69
+ - pairs_with_negatives
70
+ - positives
71
+ <!-- - **Language:** Unknown -->
72
+ <!-- - **License:** Unknown -->
73
+
74
+ ### Model Sources
75
+
76
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
77
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
78
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
79
+
80
+ ### Full Model Architecture
81
+
82
+ ```
83
+ SentenceTransformer(
84
+ (0): Transformer({'transformer_task': 'feature-extraction', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'last_hidden_state'}}, 'module_output_name': 'token_embeddings', 'architecture': 'Qwen3Model'})
85
+ (1): Pooling({'embedding_dimension': 1024, 'pooling_mode': 'lasttoken', 'include_prompt': True})
86
+ (2): Normalize({})
87
+ )
88
+ ```
89
+
90
+ ## Usage
91
+
92
+ ### Direct Usage (Sentence Transformers)
93
+
94
+ First install the Sentence Transformers library:
95
+
96
+ ```bash
97
+ pip install -U sentence-transformers
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
+ queries = [
107
+ 'Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it\nQuery: شاحن تايب سي للسيارة',
108
+ ]
109
+ documents = [
110
+ 'شاحن سيارة قرين ليون مدخلين 36 وات مع كابل تايب سي - CBK',
111
+ 'بسكويت جولون بدون سكر شكلاتة ساندوتش',
112
+ 'صوص المكرونة هاينز - 365 غ',
113
+ ]
114
+ query_embeddings = model.encode_query(queries)
115
+ document_embeddings = model.encode_document(documents)
116
+ print(query_embeddings.shape, document_embeddings.shape)
117
+ # [1, 1024] [3, 1024]
118
+
119
+ # Get the similarity scores for the embeddings
120
+ similarities = model.similarity(query_embeddings, document_embeddings)
121
+ print(similarities)
122
+ # tensor([[ 0.6144, 0.0285, -0.0279]])
123
+ ```
124
+ <!--
125
+ ### Direct Usage (Transformers)
126
+
127
+ <details><summary>Click to see the direct usage in Transformers</summary>
128
+
129
+ </details>
130
+ -->
131
+
132
+ <!--
133
+ ### Downstream Usage (Sentence Transformers)
134
+
135
+ You can finetune this model on your own dataset.
136
+
137
+ <details><summary>Click to expand</summary>
138
+
139
+ </details>
140
+ -->
141
+
142
+ <!--
143
+ ### Out-of-Scope Use
144
+
145
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
146
+ -->
147
+
148
+ <!--
149
+ ## Bias, Risks and Limitations
150
+
151
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
152
+ -->
153
+
154
+ <!--
155
+ ### Recommendations
156
+
157
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
158
+ -->
159
+
160
+ ## Training Details
161
+
162
+ ### Training Datasets
163
+
164
+ #### pairs_with_negatives
165
+
166
+ * Dataset: pairs_with_negatives
167
+ * Size: 124,261 training samples
168
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
169
+ * Approximate statistics based on the first 1000 samples:
170
+ | | anchor | positive | negative |
171
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
172
+ | type | string | string | string |
173
+ | details | <ul><li>min: 23 tokens</li><li>mean: 29.44 tokens</li><li>max: 42 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 15.95 tokens</li><li>max: 49 tokens</li></ul> | <ul><li>min: 2 tokens</li><li>mean: 14.87 tokens</li><li>max: 40 tokens</li></ul> |
174
+ * Samples:
175
+ | anchor | positive | negative |
176
+ |:--------------------------------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------|:-------------------------------------------------------------------|
177
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: واقي للوجه دهني</code> | <code>Anthelios Oil Control (Dry Touch) - La Roche Posay</code> | <code>Anthelios Invisible Mist (Dry Touch) - La Roche Posay</code> |
178
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: تن منارة زليتن</code> | <code>تن منارة زليتن بزيت دوار الشمس - 160 غ</code> | <code>تن فاني بزيت دوار الشمس - 160 غ</code> |
179
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: فانتا زجاجة صغيرة</code> | <code>مشروب فانتا برتقال زجاجة - 330 مل</code> | <code>مشروب فانتا - 1 ل (برتقال)</code> |
180
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
181
+ ```json
182
+ {
183
+ "scale": 20.0,
184
+ "similarity_fct": "cos_sim",
185
+ "mini_batch_size": 8,
186
+ "gather_across_devices": false,
187
+ "directions": [
188
+ "query_to_doc"
189
+ ],
190
+ "partition_mode": "joint",
191
+ "hardness_mode": null,
192
+ "hardness_strength": 0.0
193
+ }
194
+ ```
195
+
196
+ #### positives
197
+
198
+ * Dataset: positives
199
+ * Size: 217,800 training samples
200
+ * Columns: <code>anchor</code> and <code>positive</code>
201
+ * Approximate statistics based on the first 1000 samples:
202
+ | | anchor | positive |
203
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
204
+ | type | string | string |
205
+ | details | <ul><li>min: 23 tokens</li><li>mean: 29.61 tokens</li><li>max: 52 tokens</li></ul> | <ul><li>min: 2 tokens</li><li>mean: 13.81 tokens</li><li>max: 41 tokens</li></ul> |
206
+ * Samples:
207
+ | anchor | positive |
208
+ |:-----------------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------|
209
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: نحب جبنة القرية</code> | <code>ميرسين جبنة القرية 200 جم</code> |
210
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: كابل شحن مايكرو Moxom A2.4</code> | <code>كابل شحن مايكرو Moxom - A2.4</code> |
211
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: dry idea</code> | <code>Dry idea (powder fresh)</code> |
212
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
213
+ ```json
214
+ {
215
+ "scale": 20.0,
216
+ "similarity_fct": "cos_sim",
217
+ "mini_batch_size": 8,
218
+ "gather_across_devices": false,
219
+ "directions": [
220
+ "query_to_doc"
221
+ ],
222
+ "partition_mode": "joint",
223
+ "hardness_mode": null,
224
+ "hardness_strength": 0.0
225
+ }
226
+ ```
227
+
228
+ ### Evaluation Datasets
229
+
230
+ #### pairs_with_negatives
231
+
232
+ * Dataset: pairs_with_negatives
233
+ * Size: 1,256 evaluation samples
234
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
235
+ * Approximate statistics based on the first 1000 samples:
236
+ | | anchor | positive | negative |
237
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
238
+ | type | string | string | string |
239
+ | details | <ul><li>min: 23 tokens</li><li>mean: 29.56 tokens</li><li>max: 40 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 16.04 tokens</li><li>max: 43 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 15.05 tokens</li><li>max: 33 tokens</li></ul> |
240
+ * Samples:
241
+ | anchor | positive | negative |
242
+ |:-------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------|:--------------------------------------|
243
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: هيبوتك بوزن</code> | <code>عطر Hypnotic Poison - PERFECTO COLLECTION</code> | <code>عطر Poison Girl - Dior</code> |
244
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: شاحن مايكرو 2.4A</code> | <code>شحن مايكرو Smila - 2.4A</code> | <code>شحن تايب سي Smila - 2.4A</code> |
245
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: تمر صعيدي</code> | <code>تمر صعيدي مشفوط</code> | <code>تمر قصيم مشفوط</code> |
246
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
247
+ ```json
248
+ {
249
+ "scale": 20.0,
250
+ "similarity_fct": "cos_sim",
251
+ "mini_batch_size": 8,
252
+ "gather_across_devices": false,
253
+ "directions": [
254
+ "query_to_doc"
255
+ ],
256
+ "partition_mode": "joint",
257
+ "hardness_mode": null,
258
+ "hardness_strength": 0.0
259
+ }
260
+ ```
261
+
262
+ #### positives
263
+
264
+ * Dataset: positives
265
+ * Size: 2,200 evaluation samples
266
+ * Columns: <code>anchor</code> and <code>positive</code>
267
+ * Approximate statistics based on the first 1000 samples:
268
+ | | anchor | positive |
269
+ |:--------|:-----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
270
+ | type | string | string |
271
+ | details | <ul><li>min: 24 tokens</li><li>mean: 29.27 tokens</li><li>max: 42 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 13.61 tokens</li><li>max: 44 tokens</li></ul> |
272
+ * Samples:
273
+ | anchor | positive |
274
+ |:-------------------------------------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------|
275
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: حاملة أدوات القطط</code> | <code>حاملة أدوات القطة - AA04</code> |
276
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: كريم شمس أطفال</code> | <code>واقي شمس كريمي - Chicco</code> |
277
+ | <code>Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it<br>Query: كانديسارتان هيدروكلوروثيازيد</code> | <code>Candesartan and Hydrochlorothiazide 16mg/12.5mg</code> |
278
+ * Loss: [<code>CachedMultipleNegativesRankingLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#cachedmultiplenegativesrankingloss) with these parameters:
279
+ ```json
280
+ {
281
+ "scale": 20.0,
282
+ "similarity_fct": "cos_sim",
283
+ "mini_batch_size": 8,
284
+ "gather_across_devices": false,
285
+ "directions": [
286
+ "query_to_doc"
287
+ ],
288
+ "partition_mode": "joint",
289
+ "hardness_mode": null,
290
+ "hardness_strength": 0.0
291
+ }
292
+ ```
293
+
294
+ ### Training Hyperparameters
295
+ #### Non-Default Hyperparameters
296
+
297
+ - `per_device_train_batch_size`: 32
298
+ - `learning_rate`: 0.0001
299
+ - `num_train_epochs`: 1
300
+ - `warmup_steps`: 0.05
301
+ - `fp16`: True
302
+
303
+ #### All Hyperparameters
304
+ <details><summary>Click to expand</summary>
305
+
306
+ - `do_predict`: False
307
+ - `prediction_loss_only`: True
308
+ - `per_device_train_batch_size`: 32
309
+ - `per_device_eval_batch_size`: 8
310
+ - `gradient_accumulation_steps`: 1
311
+ - `eval_accumulation_steps`: None
312
+ - `torch_empty_cache_steps`: None
313
+ - `learning_rate`: 0.0001
314
+ - `weight_decay`: 0.0
315
+ - `adam_beta1`: 0.9
316
+ - `adam_beta2`: 0.999
317
+ - `adam_epsilon`: 1e-08
318
+ - `max_grad_norm`: 1.0
319
+ - `num_train_epochs`: 1
320
+ - `max_steps`: -1
321
+ - `lr_scheduler_type`: linear
322
+ - `lr_scheduler_kwargs`: None
323
+ - `warmup_ratio`: None
324
+ - `warmup_steps`: 0.05
325
+ - `log_level`: passive
326
+ - `log_level_replica`: warning
327
+ - `log_on_each_node`: True
328
+ - `logging_nan_inf_filter`: True
329
+ - `enable_jit_checkpoint`: False
330
+ - `save_on_each_node`: False
331
+ - `save_only_model`: False
332
+ - `restore_callback_states_from_checkpoint`: False
333
+ - `use_cpu`: False
334
+ - `seed`: 42
335
+ - `data_seed`: None
336
+ - `bf16`: False
337
+ - `fp16`: True
338
+ - `bf16_full_eval`: False
339
+ - `fp16_full_eval`: False
340
+ - `tf32`: None
341
+ - `local_rank`: -1
342
+ - `ddp_backend`: None
343
+ - `debug`: []
344
+ - `dataloader_drop_last`: False
345
+ - `dataloader_num_workers`: 0
346
+ - `dataloader_prefetch_factor`: None
347
+ - `disable_tqdm`: False
348
+ - `remove_unused_columns`: True
349
+ - `label_names`: None
350
+ - `load_best_model_at_end`: False
351
+ - `ignore_data_skip`: False
352
+ - `fsdp`: []
353
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
354
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
355
+ - `parallelism_config`: None
356
+ - `deepspeed`: None
357
+ - `label_smoothing_factor`: 0.0
358
+ - `optim`: adamw_torch_fused
359
+ - `optim_args`: None
360
+ - `group_by_length`: False
361
+ - `length_column_name`: length
362
+ - `project`: huggingface
363
+ - `trackio_space_id`: trackio
364
+ - `ddp_find_unused_parameters`: None
365
+ - `ddp_bucket_cap_mb`: None
366
+ - `ddp_broadcast_buffers`: False
367
+ - `dataloader_pin_memory`: True
368
+ - `dataloader_persistent_workers`: False
369
+ - `skip_memory_metrics`: True
370
+ - `push_to_hub`: False
371
+ - `resume_from_checkpoint`: None
372
+ - `hub_model_id`: None
373
+ - `hub_strategy`: every_save
374
+ - `hub_private_repo`: None
375
+ - `hub_always_push`: False
376
+ - `hub_revision`: None
377
+ - `gradient_checkpointing`: False
378
+ - `gradient_checkpointing_kwargs`: None
379
+ - `include_for_metrics`: []
380
+ - `eval_do_concat_batches`: True
381
+ - `auto_find_batch_size`: False
382
+ - `full_determinism`: False
383
+ - `ddp_timeout`: 1800
384
+ - `torch_compile`: False
385
+ - `torch_compile_backend`: None
386
+ - `torch_compile_mode`: None
387
+ - `include_num_input_tokens_seen`: no
388
+ - `neftune_noise_alpha`: None
389
+ - `optim_target_modules`: None
390
+ - `batch_eval_metrics`: False
391
+ - `eval_on_start`: False
392
+ - `use_liger_kernel`: False
393
+ - `liger_kernel_config`: None
394
+ - `eval_use_gather_object`: False
395
+ - `average_tokens_across_devices`: True
396
+ - `use_cache`: False
397
+ - `prompts`: None
398
+ - `batch_sampler`: batch_sampler
399
+ - `multi_dataset_batch_sampler`: proportional
400
+ - `router_mapping`: {}
401
+ - `learning_rate_mapping`: {}
402
+
403
+ </details>
404
+
405
+ ### Training Logs
406
+ | Epoch | Step | Training Loss | pairs with negatives loss | positives loss |
407
+ |:------:|:-----:|:-------------:|:-------------------------:|:--------------:|
408
+ | 0.9377 | 10025 | 0.4066 | - | - |
409
+ | 0.9400 | 10050 | 0.3446 | - | - |
410
+ | 0.9424 | 10075 | 0.4610 | - | - |
411
+ | 0.9447 | 10100 | 0.4466 | - | - |
412
+ | 0.9471 | 10125 | 0.4708 | - | - |
413
+ | 0.9494 | 10150 | 0.3631 | - | - |
414
+ | 0.9517 | 10175 | 0.4316 | - | - |
415
+ | 0.9541 | 10200 | 0.3524 | 0.4091 | 0.1566 |
416
+
417
+
418
+ ### Training Time
419
+ - **Training**: 12.5 minutes
420
+
421
+ ### Framework Versions
422
+ - Python: 3.12.13
423
+ - Sentence Transformers: 5.4.1
424
+ - Transformers: 5.0.0
425
+ - PyTorch: 2.10.0+cu128
426
+ - Accelerate: 1.13.0
427
+ - Datasets: 5.0.0
428
+ - Tokenizers: 0.22.2
429
+
430
+ ## Citation
431
+
432
+ ### BibTeX
433
+
434
+ #### Sentence Transformers
435
+ ```bibtex
436
+ @inproceedings{reimers-2019-sentence-bert,
437
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
438
+ author = "Reimers, Nils and Gurevych, Iryna",
439
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
440
+ month = "11",
441
+ year = "2019",
442
+ publisher = "Association for Computational Linguistics",
443
+ url = "https://arxiv.org/abs/1908.10084",
444
+ }
445
+ ```
446
+
447
+ #### CachedMultipleNegativesRankingLoss
448
+ ```bibtex
449
+ @misc{gao2021scaling,
450
+ title={Scaling Deep Contrastive Learning Batch Size under Memory Limited Setup},
451
+ author={Luyu Gao and Yunyi Zhang and Jiawei Han and Jamie Callan},
452
+ year={2021},
453
+ eprint={2101.06983},
454
+ archivePrefix={arXiv},
455
+ primaryClass={cs.LG}
456
+ }
457
+ ```
458
+
459
+ <!--
460
+ ## Glossary
461
+
462
+ *Clearly define terms in order to be accessible across audiences.*
463
+ -->
464
+
465
+ <!--
466
+ ## Model Card Authors
467
+
468
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
469
+ -->
470
+
471
+ <!--
472
+ ## Model Card Contact
473
+
474
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
475
+ -->
checkpoint-10200/adapter_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "prestoai/qwen3-embedding-0.6b-arabic-ecom",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": false,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 32,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.05,
22
+ "lora_ga_config": null,
23
+ "megatron_config": null,
24
+ "megatron_core": "megatron.core",
25
+ "modules_to_save": null,
26
+ "peft_type": "LORA",
27
+ "peft_version": "0.19.1",
28
+ "qalora_group_size": 16,
29
+ "r": 16,
30
+ "rank_pattern": {},
31
+ "revision": null,
32
+ "target_modules": [
33
+ "down_proj",
34
+ "v_proj",
35
+ "k_proj",
36
+ "o_proj",
37
+ "up_proj",
38
+ "q_proj",
39
+ "gate_proj"
40
+ ],
41
+ "target_parameters": null,
42
+ "task_type": "FEATURE_EXTRACTION",
43
+ "trainable_token_indices": null,
44
+ "use_bdlora": null,
45
+ "use_dora": false,
46
+ "use_qalora": false,
47
+ "use_rslora": false
48
+ }
checkpoint-10200/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84cd10fcf1754afd81b3c29030aaf051b2c7078124f4a1c47ba513e552624031
3
+ size 40419816
checkpoint-10200/chat_template.jinja ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0].role == 'system' %}
4
+ {{- messages[0].content + '\n\n' }}
5
+ {%- endif %}
6
+ {{- "# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
7
+ {%- for tool in tools %}
8
+ {{- "\n" }}
9
+ {{- tool | tojson }}
10
+ {%- endfor %}
11
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
12
+ {%- else %}
13
+ {%- if messages[0].role == 'system' %}
14
+ {{- '<|im_start|>system\n' + messages[0].content + '<|im_end|>\n' }}
15
+ {%- endif %}
16
+ {%- endif %}
17
+ {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
18
+ {%- for message in messages[::-1] %}
19
+ {%- set index = (messages|length - 1) - loop.index0 %}
20
+ {%- if ns.multi_step_tool and message.role == "user" and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}
21
+ {%- set ns.multi_step_tool = false %}
22
+ {%- set ns.last_query_index = index %}
23
+ {%- endif %}
24
+ {%- endfor %}
25
+ {%- for message in messages %}
26
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) %}
27
+ {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
28
+ {%- elif message.role == "assistant" %}
29
+ {%- set content = message.content %}
30
+ {%- set reasoning_content = '' %}
31
+ {%- if message.reasoning_content is defined and message.reasoning_content is not none %}
32
+ {%- set reasoning_content = message.reasoning_content %}
33
+ {%- else %}
34
+ {%- if '</think>' in message.content %}
35
+ {%- set content = message.content.split('</think>')[-1].lstrip('\n') %}
36
+ {%- set reasoning_content = message.content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
37
+ {%- endif %}
38
+ {%- endif %}
39
+ {%- if loop.index0 > ns.last_query_index %}
40
+ {%- if loop.last or (not loop.last and reasoning_content) %}
41
+ {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content.strip('\n') + '\n</think>\n\n' + content.lstrip('\n') }}
42
+ {%- else %}
43
+ {{- '<|im_start|>' + message.role + '\n' + content }}
44
+ {%- endif %}
45
+ {%- else %}
46
+ {{- '<|im_start|>' + message.role + '\n' + content }}
47
+ {%- endif %}
48
+ {%- if message.tool_calls %}
49
+ {%- for tool_call in message.tool_calls %}
50
+ {%- if (loop.first and content) or (not loop.first) %}
51
+ {{- '\n' }}
52
+ {%- endif %}
53
+ {%- if tool_call.function %}
54
+ {%- set tool_call = tool_call.function %}
55
+ {%- endif %}
56
+ {{- '<tool_call>\n{"name": "' }}
57
+ {{- tool_call.name }}
58
+ {{- '", "arguments": ' }}
59
+ {%- if tool_call.arguments is string %}
60
+ {{- tool_call.arguments }}
61
+ {%- else %}
62
+ {{- tool_call.arguments | tojson }}
63
+ {%- endif %}
64
+ {{- '}\n</tool_call>' }}
65
+ {%- endfor %}
66
+ {%- endif %}
67
+ {{- '<|im_end|>\n' }}
68
+ {%- elif message.role == "tool" %}
69
+ {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
70
+ {{- '<|im_start|>user' }}
71
+ {%- endif %}
72
+ {{- '\n<tool_response>\n' }}
73
+ {{- message.content }}
74
+ {{- '\n</tool_response>' }}
75
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
76
+ {{- '<|im_end|>\n' }}
77
+ {%- endif %}
78
+ {%- endif %}
79
+ {%- endfor %}
80
+ {%- if add_generation_prompt %}
81
+ {{- '<|im_start|>assistant\n' }}
82
+ {%- if enable_thinking is defined and enable_thinking is false %}
83
+ {{- '<think>\n\n</think>\n\n' }}
84
+ {%- endif %}
85
+ {%- endif %}
checkpoint-10200/config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "pytorch": "2.10.0+cu128",
4
+ "sentence_transformers": "5.4.1",
5
+ "transformers": "5.0.0"
6
+ },
7
+ "default_prompt_name": null,
8
+ "model_type": "SentenceTransformer",
9
+ "prompts": {
10
+ "document": "",
11
+ "query": "Instruct: Given an Arabic e-commerce search query, retrieve the product that best matches it\nQuery: "
12
+ },
13
+ "similarity_fn_name": "cosine"
14
+ }
checkpoint-10200/modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.base.modules.transformer.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.sentence_transformer.modules.pooling.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.sentence_transformer.modules.normalize.Normalize"
19
+ }
20
+ ]
checkpoint-10200/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2db3b34cdc384a98fe1b67a8be204f05c6706cea2eecc93693d25763480dcf14
3
+ size 81075515
checkpoint-10200/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7285cc8a6d80f05a85a4e43e55e124db67906a542d79805535ebd5b33df09f17
3
+ size 14581
checkpoint-10200/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09f0c94b0248969191a909c37b09b9f1088b3e4ca48b75f05037e523defa4db9
3
+ size 1383
checkpoint-10200/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30054077e85551dffdc2b8ed2fe895b61d880dff1e8d60a8cd8b3458f2b85202
3
+ size 1465
checkpoint-10200/sentence_bert_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "transformer_task": "feature-extraction",
3
+ "modality_config": {
4
+ "text": {
5
+ "method": "forward",
6
+ "method_output_name": "last_hidden_state"
7
+ }
8
+ },
9
+ "module_output_name": "token_embeddings"
10
+ }
checkpoint-10200/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67b23115c7cace3b6700c333da5038e66e95a9c2c71531b6953563a8c8bb1dee
3
+ size 11423967
checkpoint-10200/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "is_local": false,
9
+ "local_files_only": false,
10
+ "model_max_length": 128,
11
+ "pad_token": "<|endoftext|>",
12
+ "split_special_tokens": false,
13
+ "tokenizer_class": "Qwen2Tokenizer",
14
+ "unk_token": null
15
+ }
checkpoint-10200/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-10200/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f64c13625404e7e8b2e3825b7c2b388aa17cbed32c8e2be6731de0c81a3f24a0
3
+ size 5457