johnnyboycurtis commited on
Commit
63b3e76
·
verified ·
1 Parent(s): ac2eb28

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -0
README.md CHANGED
@@ -337,6 +337,21 @@ model-index:
337
 
338
  This is a [sentence-transformers](https://www.SBERT.net) model trained on the [msmarco](https://huggingface.co/datasets/sentence-transformers/msmarco-msmarco-distilbert-base-v3), [gooaq](https://huggingface.co/datasets/sentence-transformers/gooaq) and [natural_questions](https://huggingface.co/datasets/sentence-transformers/natural-questions) datasets. It maps sentences & paragraphs to a 384-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
339
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
340
  ## Model Details
341
 
342
  ### Model Description
 
337
 
338
  This is a [sentence-transformers](https://www.SBERT.net) model trained on the [msmarco](https://huggingface.co/datasets/sentence-transformers/msmarco-msmarco-distilbert-base-v3), [gooaq](https://huggingface.co/datasets/sentence-transformers/gooaq) and [natural_questions](https://huggingface.co/datasets/sentence-transformers/natural-questions) datasets. It maps sentences & paragraphs to a 384-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
339
 
340
+
341
+ This model is based on the wide architecture of [johnnyboycurtis/ModernBERT-small](https://huggingface.co/johnnyboycurtis/ModernBERT-small)
342
+
343
+ ```
344
+ small_modernbert_config = ModernBertConfig(
345
+ hidden_size=384, # A common dimension for small embedding models
346
+ num_hidden_layers=12, # Significantly fewer layers than the base's 22
347
+ num_attention_heads=6, # Must be a divisor of hidden_size
348
+ intermediate_size=1536, # 4 * hidden_size -- VERY WIDE!!
349
+ max_position_embeddings=1024, # Max sequence length for the model; originally 8192
350
+ )
351
+
352
+ model = ModernBertModel(modernbert_small_config)
353
+ ```
354
+
355
  ## Model Details
356
 
357
  ### Model Description