krixi-dev commited on
Commit
dbe28c4
·
verified ·
1 Parent(s): 30e1d21

Add abhinand/MedEmbed-small-v0.1 (ONNX, medical text embeddings)

Browse files

Adds an ONNX version of abhinand/MedEmbed-small-v0.1, a BERT-based embedding model fine-tuned for medical and clinical text.

Why we want it
We run a healthcare social platform . Our posts are full of medical terms that general-purpose models handle unevenly. We tested this model with Typesense auto-embedding on 2,000 real surgical posts, and it returns relevant results for medical queries

Details
- Source model: abhinand/MedEmbed-small-v0.1 (Apache 2.0)
- Embedding size: 384
- model_type: bert
- Converted with: optimum-cli export onnx --model abhinand/MedEmbed-small-v0.1 --task feature-extraction
- The export includes the sentence-embedding output (pooling inside the ONNX graph)
- model_md5 and vocab_md5 in config.json are computed from the files in this PR
- No indexing_prefix or query_prefix is needed

MedEmbed-small-v0.1/config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "model_md5": "3e91c78c4b43ea73a3204d17062e2338",
3
+ "vocab_file_name": "vocab.txt",
4
+ "vocab_md5": "64800d5d8528ce344256daf115d4965e",
5
+ "model_type": "bert"
6
+ }
MedEmbed-small-v0.1/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95dc6297fe0bd0f6999c42d319501838f0567700ac36c8ffc17c850abaaf2a0f
3
+ size 133059979
MedEmbed-small-v0.1/vocab.txt ADDED
The diff for this file is too large to render. See raw diff