feat: push custom model

README.md

@@ -20,7 +20,7 @@ from transformers import AutoModel, AutoTokenizer
 
 llm_name = "fine-tuned/medical-10-10-1-jinaai_jina-embeddings-v2-small-en-50-gpt-3.5-turbo-01_8647177611"
 tokenizer = AutoTokenizer.from_pretrained(llm_name)
-model = AutoModel.from_pretrained(llm_name)
+model = AutoModel.from_pretrained(llm_name, trust_remote_code=True)
 
 tokens = tokenizer("Your text here", return_tensors="pt")
 embedding = model(**tokens)

config.json

@@ -1,9 +1,10 @@
 {
-  "_name_or_path": "mytmp/finetuned_model",
+  "_name_or_path": "jinaai/jina-embeddings-v2-small-en",
   "architectures": [
-    "BertModel"
+    "JinaBertModel"
   ],
   "attention_probs_dropout_prob": 0.0,
+  "attn_implementation": null,
   "auto_map": {
     "AutoConfig": "jinaai/jina-bert-implementation--configuration_bert.JinaBertConfig",
     "AutoModel": "jinaai/jina-bert-implementation--modeling_bert.JinaBertModel",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5a6bbc42fac422b4c79c915b865b26b0ad2bc8c9c92087de130b5753ddd1fd88
-size 130803392
+oid sha256:f179302289dc6e06879a59993b324f9537abd5077ef376182d1719f79d39c4aa
+size 130769960

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d62aa666a5229b976cf8a686e2903a012bff6ccb2623f397d82ac05b38bb60be
+oid sha256:bf7a51df88d681236666ae6232e789723cbfad9a62effae50fdba25a04126278
 size 4719