openbmb
/

MiniCPM-Embedding-Light

Feature Extraction

sentence-transformers

Model card Files Files and versions

Kaguya-19 commited on Jan 21

Commit

321b9bb

·

verified ·

1 Parent(s): 1d0117a

Update scripts/transformers_demo.py

Files changed (1) hide show

scripts/transformers_demo.py +1 -1

scripts/transformers_demo.py CHANGED Viewed

@@ -2,7 +2,7 @@
 from transformers import AutoModel
 import torch
-model_name = "openbmb/UltraRAG-Embedding"
 model = AutoModel.from_pretrained(model_name, trust_remote_code=True, torch_dtype=torch.float16).to("cuda")
 # you can use flash_attention_2 for faster inference

 from transformers import AutoModel
 import torch
+model_name = "openbmb/MiniCPM-Embedding-Light"
 model = AutoModel.from_pretrained(model_name, trust_remote_code=True, torch_dtype=torch.float16).to("cuda")
 # you can use flash_attention_2 for faster inference