Add SetFit model
Browse files- 1_Pooling/config.json +2 -1
- README.md +4 -4
- config_sentence_transformers.json +3 -1
1_Pooling/config.json
CHANGED
|
@@ -5,5 +5,6 @@
|
|
| 5 |
"pooling_mode_max_tokens": false,
|
| 6 |
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
-
"pooling_mode_lasttoken": false
|
|
|
|
| 9 |
}
|
|
|
|
| 5 |
"pooling_mode_max_tokens": false,
|
| 6 |
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": true
|
| 10 |
}
|
README.md
CHANGED
|
@@ -161,12 +161,12 @@ preds = model("a fast , funny , highly enjoyable movie . ")
|
|
| 161 |
|
| 162 |
* The bold row denotes the saved checkpoint.
|
| 163 |
### Framework Versions
|
| 164 |
-
- Python: 3.
|
| 165 |
- SetFit: 1.0.3
|
| 166 |
-
- Sentence Transformers: 2.
|
| 167 |
- Transformers: 4.37.2
|
| 168 |
-
- PyTorch: 2.2.
|
| 169 |
-
- Datasets: 2.17.
|
| 170 |
- Tokenizers: 0.15.2
|
| 171 |
|
| 172 |
## Citation
|
|
|
|
| 161 |
|
| 162 |
* The bold row denotes the saved checkpoint.
|
| 163 |
### Framework Versions
|
| 164 |
+
- Python: 3.12.1
|
| 165 |
- SetFit: 1.0.3
|
| 166 |
+
- Sentence Transformers: 2.4.0
|
| 167 |
- Transformers: 4.37.2
|
| 168 |
+
- PyTorch: 2.2.1+cu121
|
| 169 |
+
- Datasets: 2.17.0
|
| 170 |
- Tokenizers: 0.15.2
|
| 171 |
|
| 172 |
## Citation
|
config_sentence_transformers.json
CHANGED
|
@@ -3,5 +3,7 @@
|
|
| 3 |
"sentence_transformers": "2.0.0",
|
| 4 |
"transformers": "4.7.0",
|
| 5 |
"pytorch": "1.9.0+cu102"
|
| 6 |
-
}
|
|
|
|
|
|
|
| 7 |
}
|
|
|
|
| 3 |
"sentence_transformers": "2.0.0",
|
| 4 |
"transformers": "4.7.0",
|
| 5 |
"pytorch": "1.9.0+cu102"
|
| 6 |
+
},
|
| 7 |
+
"prompts": {},
|
| 8 |
+
"default_prompt_name": null
|
| 9 |
}
|