Add SetFit model

1_Pooling/config.json

@@ -3,5 +3,8 @@
   "pooling_mode_cls_token": false,
   "pooling_mode_mean_tokens": true,
   "pooling_mode_max_tokens": false,
-  "pooling_mode_mean_sqrt_len_tokens": false
+  "pooling_mode_mean_sqrt_len_tokens": false,
+  "pooling_mode_weightedmean_tokens": false,
+  "pooling_mode_lasttoken": false,
+  "include_prompt": true
 }

README.md

@@ -1,30 +1,16 @@
 ---
+base_model: sentence-transformers/all-MiniLM-L6-v2
 library_name: setfit
+metrics:
+- accuracy
+pipeline_tag: text-classification
 tags:
 - setfit
 - sentence-transformers
 - text-classification
 - generated_from_setfit_trainer
-metrics:
-- accuracy
 widget: []
-pipeline_tag: text-classification
 inference: true
-base_model: sentence-transformers/all-MiniLM-L6-v2
-model-index:
-- name: SetFit with sentence-transformers/all-MiniLM-L6-v2
-  results:
-  - task:
-      type: text-classification
-      name: Text Classification
-    dataset:
-      name: Unknown
-      type: unknown
-      split: test
-    metrics:
-    - type: accuracy
-      value: 0.14285714285714285
-      name: Accuracy
 ---
 
 # SetFit with sentence-transformers/all-MiniLM-L6-v2
@@ -54,13 +40,6 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
-## Evaluation
-
-### Metrics
-| Label   | Accuracy |
-|:--------|:---------|
-| **all** | 0.1429   |
-
 ## Uses
 
 ### Direct Use for Inference
@@ -109,13 +88,13 @@ preds = model("I loved the spiderman movie!")
 ## Training Details
 
 ### Framework Versions
-- Python: 3.11.7
+- Python: 3.10.12
 - SetFit: 1.1.0.dev0
-- Sentence Transformers: 2.2.2
-- Transformers: 4.35.2
-- PyTorch: 2.1.1+cu121
-- Datasets: 2.14.5
-- Tokenizers: 0.15.1
+- Sentence Transformers: 3.0.1
+- Transformers: 4.42.4
+- PyTorch: 2.3.1+cu121
+- Datasets: 2.20.0
+- Tokenizers: 0.19.1
 
 ## Citation
 

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "checkpoints/step_5880/",
+  "_name_or_path": "checkpoints/step_168",
   "architectures": [
     "BertModel"
   ],
@@ -19,7 +19,7 @@
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "torch_dtype": "float32",
-  "transformers_version": "4.35.2",
+  "transformers_version": "4.42.4",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

config_sentence_transformers.json

@@ -1,7 +1,10 @@
 {
   "__version__": {
-    "sentence_transformers": "2.0.0",
-    "transformers": "4.6.1",
-    "pytorch": "1.8.1"
-  }
+    "sentence_transformers": "3.0.1",
+    "transformers": "4.42.4",
+    "pytorch": "2.3.1+cu121"
+  },
+  "prompts": {},
+  "default_prompt_name": null,
+  "similarity_fn_name": null
 }

config_setfit.json

@@ -1,4 +1,4 @@
 {
-  "normalize_embeddings": false,
-  "labels": null
+  "labels": null,
+  "normalize_embeddings": false
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b388e5755d9a52d79f92a417a294936cb542eb1fae67a2d53a1d1830a87f5896
+oid sha256:6df77805466cf85f633cc3a3e7c7ce198e06aec9df562f1839e015a8fadb7148
 size 90864192

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37f24b081ced3c130aa15c574f31ddefb4db1ec8e7d3eabb39e32dc9d34dfad9
+oid sha256:ad51248bc3cb40a47723a105e8754030d4e6d12b0dcf1a2b3411d8cbc15de684
 size 44127

tokenizer_config.json

@@ -47,7 +47,7 @@
   "do_lower_case": true,
   "mask_token": "[MASK]",
   "max_length": 128,
-  "model_max_length": 512,
+  "model_max_length": 256,
   "never_split": null,
   "pad_to_multiple_of": null,
   "pad_token": "[PAD]",