tim1900 commited on
Commit
00092d6
·
verified ·
1 Parent(s): 70a9422

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -11,7 +11,7 @@ pipeline_tag: sentence-similarity
11
 
12
  ## Introduction
13
 
14
- BertChunker is a text chunker based on BERT with a classifier head to predict the start token of chunks (for use in RAG, etc). It was finetuned on [MiniLM-L6-H384-uncased](https://huggingface.co/nreimers/MiniLM-L6-H384-uncased). The whole training lasted for 10 minutes on a Nvidia P40 GPU on a 50 MB synthetized dataset.
15
 
16
  This repo includes model checkpoint, BertChunker class definition file and all the other files needed.
17
 
 
11
 
12
  ## Introduction
13
 
14
+ BertChunker is a text chunker based on BERT with a classifier head to predict the start token of chunks (for use in RAG, etc). It was finetuned on [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2). The whole training lasted for 10 minutes on a Nvidia P40 GPU on a 50 MB synthetized dataset.
15
 
16
  This repo includes model checkpoint, BertChunker class definition file and all the other files needed.
17