Add SetFit model
Browse files- README.md +3 -3
- config.json +1 -1
- config_setfit.json +2 -2
- model_head.pkl +1 -1
README.md
CHANGED
@@ -178,10 +178,10 @@ preds = model("President Blasts Firing Notre Dame's outgoing president criticize
|
|
178 |
- Python: 3.10.13
|
179 |
- SetFit: 1.0.3
|
180 |
- Sentence Transformers: 2.7.0
|
181 |
-
- Transformers: 4.
|
182 |
-
- PyTorch: 2.
|
183 |
- Datasets: 2.18.0
|
184 |
-
- Tokenizers: 0.
|
185 |
|
186 |
## Citation
|
187 |
|
|
|
178 |
- Python: 3.10.13
|
179 |
- SetFit: 1.0.3
|
180 |
- Sentence Transformers: 2.7.0
|
181 |
+
- Transformers: 4.40.2
|
182 |
+
- PyTorch: 2.3.0+cu121
|
183 |
- Datasets: 2.18.0
|
184 |
+
- Tokenizers: 0.19.1
|
185 |
|
186 |
## Citation
|
187 |
|
config.json
CHANGED
@@ -19,7 +19,7 @@
|
|
19 |
"pad_token_id": 0,
|
20 |
"position_embedding_type": "absolute",
|
21 |
"torch_dtype": "float32",
|
22 |
-
"transformers_version": "4.
|
23 |
"type_vocab_size": 2,
|
24 |
"use_cache": true,
|
25 |
"vocab_size": 30522
|
|
|
19 |
"pad_token_id": 0,
|
20 |
"position_embedding_type": "absolute",
|
21 |
"torch_dtype": "float32",
|
22 |
+
"transformers_version": "4.40.2",
|
23 |
"type_vocab_size": 2,
|
24 |
"use_cache": true,
|
25 |
"vocab_size": 30522
|
config_setfit.json
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
{
|
2 |
-
"
|
3 |
-
"
|
4 |
}
|
|
|
1 |
{
|
2 |
+
"normalize_embeddings": false,
|
3 |
+
"labels": null
|
4 |
}
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 13191
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:34815a23dc91a292e9dc134c97d272672aea1c4264f89cf86d66cb21631c71f1
|
3 |
size 13191
|